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

例一

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)

运行结果：

例二

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)

运行结果：