赞
踩
分类目录:《深入浅出Pytorch函数》总目录
相关文章:
· 深入浅出TensorFlow2函数——tf.linalg.matvec
· 深入浅出Pytorch函数——torch.mv
· 深入浅出PaddlePaddle函数——paddle.mv
计算矩阵input
和向量vec
的乘积。
torch.mv(input, vec, *, out=None) → Tensor
input
: [Tensor
] 待被计算乘积的矩阵。vec
: [Tensor
] 待被计算乘积的向量。out
: [Tensor
] 输出的张量。>>> mat = torch.randn(2, 3)
>>> vec = torch.randn(3)
>>> torch.mv(mat, vec)
tensor([ 1.0404, -0.6361])
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。