unsqueeze
注意:1、a.unsqueeze(i), 其中i值的范围为:[-a.dim() - 1, a.dim() + 1) -> [-5, 5),左开右闭,所以当i=5时报错。
2、i值就表示新张量新增的维度,例如a.unsqueeze(0),天辰手机app就表示新向量的第0维是增加的,新向量torch.Size([1, 4, 1, 28, 28])的第0维度是增加的。
unsqueeze(i)方法可以连续使用,天辰手机版下载但要注意后一个unsqueeze是用在前一个新张量的基础上的。例如unsqueeze(2)是用于b.unsqueeze(1)[形状为torc
网友回应