python 切片功能

import torch
b = torch.rand(3,3,3)
print(b)
print(b[...,1])
print(b.shape)

b[…,0]取第一列的值，最后一维度去掉
b[…,1]取第二列中的值，最后一维度去掉

 tensor([[[0.8592, 0.4297, 0.8940],
     [0.3419, 0.7795, 0.3489],
     [0.2487, 0.1574, 0.1543]],

    [[0.0347, 0.4018, 0.6031],
     [0.2260, 0.2296, 0.5131],
     [0.3795, 0.1552, 0.6835]]
    [[0.8834, 0.0935, 0.4816],
     [0.3250, 0.5582, 0.9697],
     [0.9690, 0.1123, 0.6046]]])

     tensor([[0.4297, 0.7795, 0.1574],
    [0.4018, 0.2296, 0.1552],
    [0.0935, 0.5582, 0.1123]])

torch.Size([3, 3])

b = torch.rand(3,3,4）
print(b[:,2]) #取出第二维的值 得到为（3，4）
b(:,-1) #取出最后一维的数据```形状为[3,4]


tensor([[[0.7188, 0.6875, 0.0182, 0.9819],
         [0.1312, 0.5578, 0.9974, 0.1305],
         [0.2816, 0.4980, 0.9032, 0.2277]],

        [[0.4720, 0.3716, 0.4430, 0.4875],
         [0.9123, 0.9438, 0.1471, 0.6593],
         [0.9440, 0.9829, 0.8177, 0.9017]],

        [[0.4214, 0.5444, 0.8830, 0.9441],
         [0.9700, 0.2396, 0.7183, 0.8840],
         [0.2813, 0.2350, 0.3520, 0.6333]]])
tensor([[0.2816, 0.4980, 0.9032, 0.2277],
        [0.9440, 0.9829, 0.8177, 0.9017],
        [0.2813, 0.2350, 0.3520, 0.6333]])
torch.Size([3, 4])
tensor([[0.2816, 0.4980, 0.9032, 0.2277],
        [0.9440, 0.9829, 0.8177, 0.9017],
        [0.2813, 0.2350, 0.3520, 0.6333]])
torch.Size([3, 4])