PyTorch torch.is_complex 函数

PyTorch torch.is_complex 函数 Pytorch torch 参考手册 torch.is_complex 是 PyTorch 中用于检查张量数据类型是否为复数类型的函数。函数定义 torch.is_complex(input) 使用示例实例 [mycode4 type='python'] import torch # 检查实数张量 x = torch.tensor([1..

更新于 2026/7/16

教程详情页默认 noindex。

PyTorch torch.is_complex 函数

Pytorch torch 参考手册

torch.is_complex 是 PyTorch 中用于检查张量数据类型是否为复数类型的函数。

函数定义

torch.is_complex(input)

使用示例

实例

import torch

# 检查实数张量
x = torch.tensor([1.0, 2.0, 3.0])
print("实数:", torch.is_complex(x))

# 检查复数张量
y = torch.tensor([1+2j, 3+4j])
print("复数:", torch.is_complex(y))

输出结果为：

实数: False
复数: True

Pytorch torch 参考手册