Size ([3, 2, 4, 1, 2, 1]) print(y.shape) z = torch.unsqueeze(y,dim=0)# Add a dimension of 1 in the 0th position print(z.size()) # torch.
確定! 回上一頁