Pytorch切片后的维度记录

jggybf

已于 2022-11-08 15:31:32 修改

阅读量464

点赞数

文章标签： pytorch python 人工智能

于 2022-09-02 17:12:58 首次发布

本文链接：https://blog.csdn.net/xdaiqy/article/details/126667208

版权

import torch

pred = torch.randn(2,2,4,4)
print(pred)
print(pred.dim())
print(pred.shape)

pred1 = pred[:,1::,:,:] #dim = 4，同pred[:,1:,:,:]
print(pred1)
print(pred1.dim())
print(pred1.shape)

pred2 = pred[:,1,:,:] #dim = 3
print(pred2)
print(pred2.dim())
print(pred2.shape)


#结果
tensor([[[[-0.0425,  0.0337,  1.5198, -1.1996],
          [ 0.9822, -0.0825, -0.5892,  0.3882],
          [ 1.0140, -0.1410, -1.9937,  0.2072],
          [-0.5766, -0.7825, -1.4023, -1.1049]],

         [[ 1.8145, -0.6655,  0.5753,  1.6547],
          [-0.3921, -0.7351,  0.6879, -1.9041],
          [ 0.5582, -0.8010, -1.0906, -1.1293],
          [-0.9992, -1.2167, -1.3798, -0.1502]]],


        [[[ 1.2841, -1.5544, -0.9725, -1.4550],
          [-0.4082,  1.7195, -1.1443, -0.7785],
          [ 0.5337,  0.7424, -1.1704,  0.9458],
          [ 0.9269, -1.9151,  1.0441, -0.2150]],

         [[-1.0973,  1.6943, -1.6763,  0.4528],
          [-0.7315,  0.8701,  2.3759,  0.5036],
          [-0.2928, -0.4785, -0.3281, -0.2773],
          [-0.8616, -1.5998, -0.1990,  0.7271]]]])
4
torch.Size([2, 2, 4, 4])
tensor([[[[ 1.8145, -0.6655,  0.5753,  1.6547],
          [-0.3921, -0.7351,  0.6879, -1.9041],
          [ 0.5582, -0.8010, -1.0906, -1.1293],
          [-0.9992, -1.2167, -1.3798, -0.1502]]],


        [[[-1.0973,  1.6943, -1.6763,  0.4528],
          [-0.7315,  0.8701,  2.3759,  0.5036],
          [-0.2928, -0.4785, -0.3281, -0.2773],
          [-0.8616, -1.5998, -0.1990,  0.7271]]]])
4
torch.Size([2, 1, 4, 4])
tensor([[[ 1.8145, -0.6655,  0.5753,  1.6547],
         [-0.3921, -0.7351,  0.6879, -1.9041],
         [ 0.5582, -0.8010, -1.0906, -1.1293],
         [-0.9992, -1.2167, -1.3798, -0.1502]],

        [[-1.0973,  1.6943, -1.6763,  0.4528],
         [-0.7315,  0.8701,  2.3759,  0.5036],
         [-0.2928, -0.4785, -0.3281, -0.2773],
         [-0.8616, -1.5998, -0.1990,  0.7271]]])
3
torch.Size([2, 4, 4])

Process finished with exit code 0