torch.permute可以进行多个维度的交换,而torch.transpose只能进行两个维度的交换. In [1]: dat = torch.tensor( [[[ 1, 2, 3, 4], [ 5, 6, 7, 8], ...
確定! 回上一頁