第45个方法
torch.transpose(input, dim0, dim1) → Tensor
返回一个tensor,此tensor是输入tensor的转置版本,给定的维度dim0和dim1将会被交换。
结果tensor和原tensor共享内存,所以改变其中一个的内容也会改变另外一个里面的内容。
参数介绍:
input
:输入tensor。dim0(int)
:要转换的第一个维度。- ``dim1(int):要转换的第二个维度。
实例如下所示:
>>> x = torch.randn(2, 3)
>>> x
tensor([[ 1.0028, -0.9893, 0.5809],
[-0.1669, 0.7299, 0.4942]])
>>> torch.transpose(x, 0, 1)
tensor([[ 1.0028, -0.1669],
[-0.9893, 0.7299],
[ 0.5809, 0.4942]])