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
pytorch 简单公式
最新推荐文章于 2024-07-11 10:30:00 发布