How to check if an object is a PyTorch Tensor?

To check if an object is a tensor or not, we can use the torch.is_tensor() method. It returns True if the input is a tensor; False otherwise.

Syntax

torch.is_tensor(input)

Parameters

• input – The object to be checked, if it is a tensor or not .

Output

It returns True if the input is a tensor; else False.

Steps

• Import the required library. The required library is torch.

• Define a tensor or other object.

• Check if the created object is a tensor or not using torch.is_tensor(input).

• Display the result.

Example 1

# import the required library
import torch

# create an object x
x = torch.rand(4)
print(x)

# check if the above created object is a tensor
print(torch.is_tensor(x))

Output

tensor([0.9270, 0.2194, 0.2078, 0.5716])
True

Example 2

# import the required library
import torch

# define an object x
x = 4

# check if the above created object is a tensor
if torch.is_tensor(x):
   print ("The input object is a Tensor.")
else:
   print ("The input object is not a Tensor.")

Output

The input object is not a Tensor.

In Example 2, torch.is_tensor(x) returns False, hence the input object is not a Tensor.