torch.ones理解
input = torch.ones(1, 1, 5, 5)input = Variable(input)输出tensor([[[[1., 1., 1., 1., 1.],[1., 1., 1., 1., 1.],[1., 1., 1., 1., 1.],[1., 1., 1., 1., 1.],[1., 1., 1., 1., 1.]]]])调整括号[[[
·
input = torch.ones(1, 1, 5, 5)
input = Variable(input)
输出
tensor([[[[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.]]]])
调整括号
[
[
[
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.]
]
]
]
理解:是从后往前看的,最后两个是行列二维矩阵,前面的都是个数。例如(1, 1, 5, 5)是1个(1, 5, 5)的矩阵,(1, 5,
5)又可以理解为1个(5, 5)的矩阵,到二维截至。
更多推荐
所有评论(0)