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

PyTorch torch.view_as_real 函数


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

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)

输出结果为:

原始复数张量形状: torch.Size([3])
实数张量形状: torch.Size([3, 2])
实数张量:
tensor([[1., 2.],
        [3., 4.],
        [5., 6.]])

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