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对应的矩阵内容为
经过如下的对应的矩阵变换
X = X.transpose(0,1)
变换的结果为
这里可以将第一个矩阵假设为X,第二个矩阵假设为Y,第三个矩阵假设为Z,第四个矩阵假设为W,矩阵变换的过程就为由
[[X,Y],
[Z,W]]
变换为对应的
[[X,Z]
[Y,W]]
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)