Función torch.is_complex de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.is_complexEs una función en PyTorch que se utiliza para comprobar si el tipo de datos de un tensor es de tipo complejo.

Definición de la función

torch.is_complex(input)

Ejemplo de uso

Ejemplo

import torch

# Comprobar tensor real
x = torch.tensor([1.0, 2.0, 3.0])
print("Real:", torch.is_complex(x))

# Comprobar tensor complejo
y = torch.tensor([1+2j, 3+4j])
print("Complejo:", torch.is_complex(y))

El resultado de salida es:

实数: False
复数: True

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones