pytorch transpose函数解析

transpose函数的本意是将两个相应的维度进行变换,比如如下的这个tensor矩阵

import  torch
X = torch.Tensor([[1,2,3],[4,5,6]])
X = X.transpose(0, 1)
print(X)

输出结果为
对应输出结果
可以看出矩阵相当于一个变换操作,原先的行变为现在的列,原先的列变为现在的行
现在变换X对应的矩阵的内容,将X变为227的对应的矩阵
原先X对应的矩阵内容为
pytorch transpose函数解析_第1张图片
经过如下的对应的矩阵变换

X = X.transpose(0,1)

变换的结果为
pytorch transpose函数解析_第2张图片这里可以将第一个矩阵假设为X,第二个矩阵假设为Y,第三个矩阵假设为Z,第四个矩阵假设为W,矩阵变换的过程就为由
[[X,Y],
[Z,W]]
变换为对应的
[[X,Z]
[Y,W]]

你可能感兴趣的:(pytorch笔记,nlp学习笔记)