PyTorch torch.view_as_complex 函数
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)
# 创建实数张量,最后一维必须有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 参考手册