PyTorch torch.is_storage 函数
Pytorch torch 参考手册torch.is_storage 是 PyTorch 中用于检查对象是否为存储对象的函数。
函数定义
torch.is_storage(obj)
使用示例
实例
import torch
# 创建张量
x = torch.tensor([1, 2, 3])
# 检查是否为存储对象
print(torch.is_storage(x))
print(torch.is_storage(x.storage()))
# 创建张量
x = torch.tensor([1, 2, 3])
# 检查是否为存储对象
print(torch.is_storage(x))
print(torch.is_storage(x.storage()))
输出结果为：
False True