In[18]: a.unsqueeze(0).shape # 在0的前面插入一个维度. Out[18]: torch.Size([1, 4, 1, 28, 28]) # 理解上就是在batch的基础上增加了组。
確定! 回上一頁