A = t.ones(3, 4) # torch.tensor -> numpy.ndarray B = A.numpy() ... 在指定位置增加新的维度-> unsqueeze In [22]: A.shape Out[22]: torch.
確定! 回上一頁