PyTorch torch.view_as_real 函数
torch.view_as_real 是 PyTorch 中用于将复数张量视图转换为实数张量的函数。它将每个复数值(实部和虚部)拆分为两个单独的值,最后一维将有两个元素。
函数定义
torch.view_as_real(input)
使用示例
实例
import torch
# 创建复数张量
x = torch.tensor([1+2j, 3+4j, 5+6j])
# 转换为实数视图
y = torch.view_as_real(x)
print("原始复数张量形状:", x.shape)
print("实数张量形状:", y.shape)
print("实数张量:")
print(y)
# 创建复数张量
x = torch.tensor([1+2j, 3+4j, 5+6j])
# 转换为实数视图
y = torch.view_as_real(x)
print("原始复数张量形状:", x.shape)
print("实数张量形状:", y.shape)
print("实数张量:")
print(y)
输出结果为:
原始复数张量形状: torch.Size([3])
实数张量形状: torch.Size([3, 2])
实数张量:
tensor([[1., 2.],
[3., 4.],
[5., 6.]])

Pytorch torch 参考手册