每天五分钟深度学习框架pytorch:pytorch中的线性代数常见的操作
diag
diag的作用是取一个矩阵的对角线元素为向量,或者取向量作为对角阵的对角线元素
a=torch.randn(3,3)
b=torch.diag(a)
c=torch.diag(b)
print(a)
print(b)
print(c)
转置t()
a=torch.randn(3,3)
b=a.t()
print(a)
print(b)
点乘
x.mul(y)或者x*y,二者都具有广播能力,要求x,y矩阵分别满足(i,n),(i,n)或者(i,n)(i,1)
点积
相同维度(只能是一维)的矩阵对位相乘,然后再相加,结果是一个标量,表示二者的相似度
原文地址:https://blog.csdn.net/huanfeng_AI/article/details/140594403
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!