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


Python PyTorch view_as_complex用法及代码示例


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

用法:

torch.view_as_complex(input) → Tensor

参数

input(Tensor) -输入张量。

input 的视图作为复数张量返回。对于 size 的输入复数张量,此函数返回 size 的新复数张量,其中输入张量的最后一个维度应表示复数的实部和虚部。

警告

view_as_complex() 仅支持具有 torch.dtype torch.float64torch.float32 的张量。输入的最后一个维度预计为 size 2。此外,张量的最后一个维度的 stride 必须为 1。所有其他维度的步幅必须是偶数。

例子:

>>> x=torch.randn(4, 2)
>>> x
tensor([[ 1.6116, -0.5772],
        [-1.4606, -0.9120],
        [ 0.0786, -1.7497],
        [-0.6561, -1.6623]])
>>> torch.view_as_complex(x)
tensor([(1.6116-0.5772j), (-1.4606-0.9120j), (0.0786-1.7497j), (-0.6561-1.6623j)])

相关用法


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