python-pytorchHow do I access the value of a tensor in PyTorch?
To access the value of a tensor in PyTorch, you can call the .item()
method on the tensor object. This method will return the value of the tensor as a Python object.
For example,
import torch
x = torch.tensor([[1, 2], [3, 4]])
print(x.item())
Output example
1
The .item()
method works for scalar tensors, i.e., tensors with only one element. If the tensor has multiple elements, then you can use the .numpy()
method to convert the tensor to a NumPy array and then access the elements using array indexing.
For example,
import torch
x = torch.tensor([[1, 2], [3, 4]])
print(x.numpy()[1,1])
Output example
4
You can also use the .tolist()
method to convert the tensor to a Python list and then access the elements using list indexing.
For example,
import torch
x = torch.tensor([[1, 2], [3, 4]])
print(x.tolist()[1][1])
Output example
4
For more information, see the PyTorch documentation.
More of Python Pytorch
- What is the most compatible version of Python to use with PyTorch?
- How do I uninstall Python PyTorch?
- How do I check the version of Python and PyTorch I am using?
- How do I use PyTorch with Python version 3.11?
- How can I use Python and PyTorch to change the shape of a tensor?
- How do I save a PyTorch tensor to a file using Python?
- How can I use PyTorch with Python 3.9?
- How do I check the Python version requirements for PyTorch?
- How can I use Yolov5 with PyTorch?
See more codes...