permute (*dims) → Tensor. 使用 permute 能够一次性调整多个维度 >>> import torch >>> a = torch.arange(24).reshape(2, 3, 4) >>> a.shape torch.
確定! 回上一頁