torch.is_tensor — PyTorch 2.7 documentation (original) (raw)

• Docs >
• torch >
• torch.is_tensor
• 

Shortcuts

torch.is_tensor(obj, /)[source][source]¶

Returns True if obj is a PyTorch tensor.

Note that this function is simply doing isinstance(obj, Tensor). Using that isinstance check is better for typechecking with mypy, and more explicit - so it’s recommended to use that instead ofis_tensor.

Parameters

obj (object) – Object to test

Return type

typing_extensions.TypeIs[Tensor]

Example:

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

• torch.is_tensor
  • is_tensor()