pytorch中的矩阵变换

交换维度位置

1.transpose()

函数返回输入矩阵input的转置。交换维度dim0和dim1

torch.transpose(input, dim0, dim1, out=None) → Tensor

##交换0和1维度
x = x.transpose(0,1)

x = x.transpose(0,1)
##等价于
x.transpose_()
# 相当于x = x + 1 简化为 x+=1

2.permute()

将tesor维度换位

permute(dims) → Tensor


x = torch.randn(2,3,4)
##按照新维度生成 (3,4,2)
x = x.permute(1,2,0)

transpose和permute

矩阵reshape

1.view()

# 使用方法
x_tensor.view(*args) → Tensor


x = torch.randn(4, 4) 
x.view(2,8) 

2.reshape()

于view相同

3.区别

view()是共享内存，reshape()不共享内存

view和reshape