矩阵乘法,矩阵中各元素对应相乘及其梯度计算;

矩阵乘法:x=([x1,x2]) w=([w1,w2],[w3,w4])
torch.matmul(x,w)=([x1w1+x2w3,x1w2+x2+w4])
x.grad=(w1+w2,w3+w4)
矩阵中各元素对应相乘
x=([x1,x2]) w=([w1,w2],[w3,w4])
x * w=([x1
w1,x2w2],[x1w3,x2w4])
(x
w在pytorch中为矩阵各个元素对应相乘)
x.grad=(w1+w3,w2+w4)
代码x*y

x=torch.tensor([1.,2.],requires_grad=True)
y=torch.tensor([[1.,2.],[3.,4.]])
z=x*y
print(z)
z.backward(torch.ones_like(z))
print(x.grad)

运行:

tensor([[1., 4.],
        [3., 8.]], grad_fn=<MulBackward0>)
tensor([4., 6.])

代码矩阵相乘

x=torch.tensor([1.,2.],requires_grad=True)
y=torch.tensor([[1.,2.],[3.,4.]])
z=torch.matmul(x,y)
print(z)
z.backward(torch.ones_like(z))
print(x.grad)

运行结果:

tensor([ 7., 10.], grad_fn=<SqueezeBackward3>)
tensor([3., 7.])
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值