[Torch] 2. torch

来源:互联网 发布:tensorflow anaconda 编辑:程序博客网 时间:2024/05/21 08:36
  • 声明一个tensor:t = torch.Tensor(2, 3, 4),其结果是生成一个2层3行4列的张量,其元素未初始化

  • 张量之间的赋值是复制引用,即重名;要生成一份拷贝,需要调用:clone方法,即s = t:clone()

  • 声明一个向量可以用v = torch.Tensor{ 1, 2, 3, 4 }

  • 声明一个全为一的矩阵(类似于MATLAB):torch.ones(N)

  • 定义一个矩阵:m = torch.Tensor{ { 1, 2, 3 }, { 4, 5, 6 } }

  • 截取一个子矩阵:

      m = torch.Tensor{ { 1, 2, 3 }, { 4, 5, 6 } }      --[[      1 2 3      4 5 6      --]]      m[{ {1, 2}, {1} }]      m[{ {}, 1 }]      --[[       1      4      --]]
0 0
原创粉丝点击