torch.masked.maskedtensor.core.is_masked_tensor

torch.masked.maskedtensor.core.is_masked_tensor(obj, /) [исходный код]

Возвращает True, если входные данные являются MaskedTensor, иначе False

Параметры:

a – любые входные данные

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

TypeIs[MaskedTensor]

Примеры

>>> from torch.masked import MaskedTensor
>>> data = torch.arange(6).reshape(2, 3)
>>> mask = torch.tensor([[True, False, False], [True, True, False]])
>>> mt = MaskedTensor(data, mask)
>>> is_masked_tensor(mt)
True