0
点赞
收藏
分享

微信扫一扫

Linux系统下的DevOps工具链:从CI到CD

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)

点积

相同维度(只能是一维)的矩阵对位相乘,然后再相加,结果是一个标量,表示二者的相似度

举报

相关推荐

0 条评论