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("Real:", torch.is_complex(x))
# Проверить комплексный тензор
y = torch.tensor([1+2j, 3+4j])
print("Complex:", torch.is_complex(y))
# Проверить вещественный тензор
x = torch.tensor([1.0, 2.0, 3.0])
print("Real:", torch.is_complex(x))
# Проверить комплексный тензор
y = torch.tensor([1+2j, 3+4j])
print("Complex:", torch.is_complex(y))
Результат вывода:
实数: False 复数: True
Другие расширения