常见API
- view / reshape
- squeeze / unsqueeze
- transpose / t / permute
- expand / repeat
Squeeze / unsqueeze
- unsqueeze
注意：1、a.unsqueeze(i), 其中i值的范围为：[-a.dim() - 1, a.dim() + 1) -> [-5, 5)，左开右闭，所以当i=5时报错。
2、i值就表示新张量新增的维度，例如a.unsqueeze(0)，就表示新向量的第0维是增加的，新向量torch.Size([1, 4, 1, 28, 28])的第0维度是增加的。
unsqueeze(i)方法可以连续使用，但要注意后一个unsqueeze是用在前一个新张量的基础上的。例如unsqueeze(2)是用于b.unsqueeze(1)[形状为torch.Size([32, 1])， 维度为2]上的，而不是用于b[形状为torch.Size([32])， 维度为1]上的。
-
squeeze
网友评论