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


Python PyTorch cross用法及代码示例


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

用法:

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

参数

  • input(Tensor) -输入张量。

  • other(Tensor) -第二个输入张量

  • dim(int,可选的) -cross-product 的尺寸。

关键字参数

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

返回 inputother 的维度 dim 中向量的叉积。

inputother的大小必须相同,并且它们的dim维度的大小应该是3。

如果未给出dim,则默认为找到的第一个尺寸为 3 的维度。请注意,这可能是意外的。

例子:

>>> a = torch.randn(4, 3)
>>> a
tensor([[-0.3956,  1.1455,  1.6895],
        [-0.5849,  1.3672,  0.3599],
        [-1.1626,  0.7180, -0.0521],
        [-0.1339,  0.9902, -2.0225]])
>>> b = torch.randn(4, 3)
>>> b
tensor([[-0.0257, -1.4725, -1.2251],
        [-1.1479, -0.7005, -1.9757],
        [-1.3904,  0.3726, -1.1836],
        [-0.9688, -0.7153,  0.2159]])
>>> torch.cross(a, b, dim=1)
tensor([[ 1.0844, -0.5281,  0.6120],
        [-2.4490, -1.5687,  1.9792],
        [-0.8304, -1.3037,  0.5650],
        [-1.2329,  1.9883,  1.0551]])
>>> torch.cross(a, b)
tensor([[ 1.0844, -0.5281,  0.6120],
        [-2.4490, -1.5687,  1.9792],
        [-0.8304, -1.3037,  0.5650],
        [-1.2329,  1.9883,  1.0551]])

相关用法


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