- Data Structure
- Networking
- RDBMS
- Operating System
- Java
- MS Excel
- iOS
- HTML
- CSS
- Android
- Python
- C Programming
- C++
- C#
- MongoDB
- MySQL
- Javascript
- PHP
- Physics
- Chemistry
- Biology
- Mathematics
- English
- Economics
- Psychology
- Social Studies
- Fashion Studies
- Legal Studies
- Selected Reading
- UPSC IAS Exams Notes
- Developer's Best Practices
- Questions and Answers
- Effective Resume Writing
- HR Interview Questions
- Computer Glossary
- Who is Who
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.
Advertisements