[...,1],[...,0]

瞎写的，大家不必当真哈

已于 2022-03-26 10:52:49 修改

阅读量97

点赞数

文章标签： python

于 2022-03-26 10:48:50 首次发布

本文链接：https://blog.csdn.net/qq_44198647/article/details/123752001

版权

a=torch.rand((2,2,2))
print(a)

tensor([[[0.6601, 0.5124],
         [0.1438, 0.2335]],

        [[0.5657, 0.0586],
         [0.0366, 0.1247]]])


print(a[...,0])


tensor([[0.6601, 0.1438],
        [0.5657, 0.0366]])

print(a[...,1])
tensor([[0.5124, 0.2335],
        [0.0586, 0.1247]])

a=torch.rand((2,1,2))
print(a)

tensor([[[0.7589, 0.7766]],

        [[0.2806, 0.7683]]])


print(a[...,0])

tensor([[0.7589],
        [0.2806]])



print(a[...,1])
tensor([[0.7766],
        [0.7683]])