pytorch 简单公式

x = torch.Tensor(3,4)
i = 0 
x:apply(function()i = i+1 return i end)
--[[
x 为
  1   2   3   4
  5   6   7   8
  9  10  11  12
]]
selected = x:select(1,2)  --第一维的第二个。就是第二行。相当于x[2]
narrowed = x:narrow(2,1,2)
--[[
th> narrowed
  1   2
  5   6
  9  10
]]
subbed = x:sub(1,3,2,3)
--[[ 一维1到3为止，二维2到3为止。
th> subbed
  2   3
  6   7
 10  11
]]
x[{1,2}] -- 2
x[{1,{1,2}}] --1 2