现在位置: 首页 > PyTorch 教程 > 正文

PyTorch torch.view_as_complex 函数


Pytorch torch 参考手册 Pytorch torch 参考手册

torch.view_as_complex 是 PyTorch 中用于将实数张量视图转换为复数张量的函数。它将最后一个维度从两个实数值(实部和虚部)转换为单个复数值。

函数定义

torch.view_as_complex(input)

使用示例

实例

import torch

# 创建实数张量,最后一维必须有2个元素
x = torch.randn(3, 2)

# 转换为复数视图
y = torch.view_as_complex(x)

print("原始实数张量形状:", x.shape)
print("复数张量形状:", y.shape)
print("复数张量:", y)

输出结果为:

原始实数张量形状: torch.Size([3, 2])
复数张量形状: torch.Size([3])
复数张量: tensor([ 1.2345+0.5678j, -0.3456+1.2345j,  2.3456-0.9876j])

Pytorch torch 参考手册 Pytorch torch 参考手册