概述
import torch
a=torch.Tensor([[1,2,3,4,5],[6,7,8,9,10]])
b=torch.Tensor([[2,4,6,8,10],[1,3,5,7,9]])
逐元素相乘 element-wise multiplication
#element-wise multiplication
c=a*b
print('c.size:t',c.size(),'nc:t',c)
输出：
c.size: torch.Size([2, 5])
c: tensor([[ 2., 8., 18., 32., 50.],
[ 6., 21., 40., 63., 90.]])
点乘dot product
#dot product
d=torch.sum(a*b,0)
e=torch.sum(a*b,1)
print('d.size:t',d.size(),'nd:t',d)
print('e.size:t',e.size(),'ne:t',e)
输出
d.size: torch.Size([5])
d: tensor([ 8., 29., 58., 95., 140.])
e.size: torch.Size([2])
e: tensor([110., 220.])
最后
以上就是碧蓝冬日为你收集整理的点乘与逐元素相乘的全部内容，希望文章能够帮你解决点乘与逐元素相乘所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复