当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch vdot用法及代码示例


本文简要介绍python语言中 torch.vdot 的用法。

用法:

torch.vdot(input, other, *, out=None) → Tensor

参数

  • input(Tensor) -点积中的第一个张量,必须是一维的。如果它很复杂,则使用它的共轭。

  • other(Tensor) -点积中的第二个张量,必须是一维的。

关键字参数

out(Tensor,可选的) -输出张量。

计算两个一维张量的点积。 vdot(a, b) 函数处理复数的方式与 dot(a, b) 不同。如果第一个参数是复数,则第一个参数的复共轭用于计算点积。

注意

与 NumPy 的 vdot 不同,torch.vdot 有意仅支持计算具有相同元素数量的两个一维张量的点积。

例子:

>>> torch.vdot(torch.tensor([2, 3]), torch.tensor([2, 1]))
tensor(7)
>>> a = torch.tensor((1 +2j, 3 - 1j))
>>> b = torch.tensor((2 +1j, 4 - 0j))
>>> torch.vdot(a, b)
tensor([16.+1.j])
>>> torch.vdot(b, a)
tensor([16.-1.j])

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.vdot。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。