本文主要是介绍torch.dot,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
torch.dot(input, other, *, out=None) → Tensor
计算两个1维张量的点积。
与Numpy的点不同,torch.dot只支持计算两个元素数量相同的1维张量的点积。
input(张量)–点积中的第一个张量,必须是1维。
other(张量)-点积中的第二个张量,必须是1维。
out(张量,可选)–输出张量。
>>> torch.dot(torch.tensor([2, 3]), torch.tensor([2, 1]))
tensor(7)
这篇关于torch.dot的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!