当前位置:   article > 正文

深入浅出Pytorch函数——torch.mv

torch.mv

分类目录:《深入浅出Pytorch函数》总目录
相关文章:
· 深入浅出TensorFlow2函数——tf.linalg.matvec
· 深入浅出Pytorch函数——torch.mv
· 深入浅出PaddlePaddle函数——paddle.mv


计算矩阵input和向量vec的乘积。

语法
torch.mv(input, vec, *, out=None) → Tensor
  • 1
参数
  • input: [Tensor] 待被计算乘积的矩阵。
  • vec: [Tensor] 待被计算乘积的向量。
  • out: [Tensor] 输出的张量。
实例
>>> mat = torch.randn(2, 3)
>>> vec = torch.randn(3)
>>> torch.mv(mat, vec)
tensor([ 1.0404, -0.6361])
  • 1
  • 2
  • 3
  • 4
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/羊村懒王/article/detail/540823
推荐阅读
相关标签
  

闽ICP备14008679号