torch.transpose(x,dim_1,dim_2)表示将x这个tensor的dim_1和dim_2进行交换,下面举个例子来说明怎么使用这个torch.transpose函数。 In [1]: import torch In [2]: ...
確定! 回上一頁