torch.is_tensor

torch.is_tensor(obj, /) [исходный код]

Возвращает True, если obj является тензором PyTorch.

Параметры:

obj (object) – Объект для проверки

Тип возвращаемого значения:

TypeIs[Tensor]

Пример:

>>> x = torch.tensor([1, 2, 3])
>>> torch.is_tensor(x)
True