网络编程
位置:首页>> 网络编程>> Python编程>> PyTorch 对应点相乘、矩阵相乘实例

PyTorch 对应点相乘、矩阵相乘实例

作者:Mr.Q  发布时间:2021-10-23 16:33:29 

标签:PyTorch,对应点,相乘,矩阵

一,对应点相乘,x.mul(y) ,即点乘操作,点乘不求和操作,又可以叫作Hadamard product;点乘再求和,即为卷积


data = [[1,2], [3,4], [5, 6]]
tensor = torch.FloatTensor(data)

tensor
Out[27]:
tensor([[ 1., 2.],
   [ 3., 4.],
   [ 5., 6.]])

tensor.mul(tensor)
Out[28]:
tensor([[ 1.,  4.],
   [ 9., 16.],
   [ 25., 36.]])

二,矩阵相乘,x.mm(y) , 矩阵大小需满足: (i, n)x(n, j)


tensor
Out[31]:
tensor([[ 1., 2.],
   [ 3., 4.],
   [ 5., 6.]])

tensor.mm(tensor.t()) # t()是转置
Out[30]:
tensor([[ 5., 11., 17.],
   [ 11., 25., 39.],
   [ 17., 39., 61.]])

来源:https://blog.csdn.net/jizhidexiaoming/article/details/82502724

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com