PyTorch 中permute的用法详解permute(dims) 将tensor的维度换位. 参数:参数是一系列的整数,代表原来张量的维度.比如三维就有0,1,2这些dimension.
確定! 回上一頁