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