自学内容网 自学内容网

每天五分钟深度学习框架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)!