python张量下标操作维度小于张量维度的情况

例一

?

1 2 3 4 5 6 7 8

import torch import torch.nn as nn   a = torch.rand([3, 3])  # a是2维的 b = a[:2]  # 对a进行切片操作时，只指明了一维。这时的意思是，只对a的第一维进行切片操作，第二维不变，相当于b=a[:2, :]   print('a:\n', a) print('b:\n', b)

运行结果：

 

例二

?

1 2 3 4 5 6 7 8

import torch import torch.nn as nn   a = torch.rand([3, 3, 3])  # a是3维的 b = a[0, :2]  # 对a进行切片操作时，只指明了前2维。这时的意思是，只对a的前2维进行切片操作，第3维不变，相当于b=a[0, :2, :]   print('a:\n', a) print('b:\n', b)

运行结果：