PyTorch torch.is_storage Function
Pytorch torch Reference Manual
torch.is_storageIt is a function in PyTorch used to check whether an object is a storage object.
Function Definition
torch.is_storage(obj)
Usage Example
Example
import torch
# Create tensor
x = torch.tensor([1, 2, 3])
# Check whether it is a storage object
print(torch.is_storage(x))
print(torch.is_storage(x.storage()))
# Create tensor
x = torch.tensor([1, 2, 3])
# Check whether it is a storage object
print(torch.is_storage(x))
print(torch.is_storage(x.storage()))
The output result is:
False True
Other Extensions