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.]]]])

调整括号

[
    [
        [
            [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）的矩阵，到二维截至。