PyTorch torch.imag 函数

PyTorch torch.imag 函数 Pytorch torch 参考手册 torch.imag 是 PyTorch 中用于获取复数张量虚部（imaginary part）的函数。函数定义 torch.imag(input) 使用示例实例 [mycode4 type='python'] import torch # 创建复数张量 x = torch.complex(torch.tens..

Updated 7/16/2026

Tutorial detail pages default to noindex.

PyTorch torch.imag 函数

Pytorch torch 参考手册

torch.imag 是 PyTorch 中用于获取复数张量虚部（imaginary part）的函数。

函数定义

torch.imag(input)

使用示例

实例

import torch

# 创建复数张量
x = torch.complex(torch.tensor([1.0, 2.0]), torch.tensor([3.0, 4.0]))

# 获取虚部
y = torch.imag(x)
print("虚部:", y)

输出结果为：

虚部: tensor([3., 4.])

Pytorch torch 参考手册