Size ([4, 1, 28, 28]) #将后3维合并prin… ... a=torch.rand(4,1,28,28); print(a.shape); print(a.unsqueeze(0).shape) # 在0号维度位置插入一个维度 ...
確定! 回上一頁