pytorch view()、transpose()和permute()的区别
transpose()和permute()的不同1、torch.transpose()是交换指定的两个维度的内容,permute()则可以一次性交换多个维度,代码示例如下a = torch.tensor([[[1, 2, 3, 4], [4, 5, 6, 7]], [[7, 8, 9, 10], [10, 11, 12, 13]], [[13, 14, 15, 16], [17, 18, 19, 20]]])print(a, a.shape)结果输出:tensor([[[ 1, 2, 3,
原创
2021-07-05 23:40:06 ·
484 阅读 ·
3 评论