pytorch的各种乘法操作,点乘和叉乘

点乘

相应点相乘,x.mul(y) ,即点乘操作,点乘不求和操作,又可以叫作Hadamard product

import torch
x = torch.tensor([[3,3],[3,3]])
y = x*x 
 #x.dot(x)
z = torch.mul(x,x) 
 #x.mul(x)
print(y)
print(z)

显示结果:

tensor([[9, 9], [9, 9]])

tensor([[9, 9], [9, 9]])



叉乘
就是传统的矩阵乘法,有三种写法

import torch
x = torch.tensor([[3,3],[3,3]])
y = torch.mm(x,x)   #等效于 y=x.matmul(x)
 #x.mm(x)
z=x@x
print(y)
print(z)

显示结果:

tensor([[18, 18], [18, 18]])

tensor([[18, 18], [18, 18]])

 

 

你可能感兴趣的:(ML)