torch.mul(a, b) 是矩阵a和b对应位相乘,a和b的维度必须相等。
<(a, b) 是矩阵a和b矩阵相乘
import torcha = torch.rand(1, 2)
b = torch.rand(1, 2)
c = torch.rand(2, 3)print(torch.mul(a, b)) # 返回 1*2 的tensor
(a, c)) # 返回 1*3 的tensor
print(torch.mul(a, c)) # 由于a、b维度不同,报错
本文发布于:2024-01-30 13:46:17,感谢您对本站的认可!
本文链接:https://www.4u4v.net/it/170659358320425.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
留言与评论(共有 0 条评论) |