PyTorch 两大转置函数 transpose() 和 permute(),
摘要:
在pytorch中转置用的函数就只有这两个 transpose() permute() transpose() torch.transpose(input, dim0, dim1, out=None) → Tensor 函数返回输入矩阵input的转置。交换维度dim0和dim1 参数: input 阅读全文
posted @ 2020-08-22 23:03 cltt 阅读(15486) 评论(0) 推荐(2)
浙公网安备 33010602011771号