torch.is_storage()
torch.is_storage()
判断给定的对象是否是一个 PyTorch 存储对象
- PyTorch 存储对象：PyTorch 中，存储对象（Storage）是一个低级别的对象，它表示一个存储数据的连续内存块。存储对象不包含任何关于数据如何解释的信息，例如数据类型或形状。它仅仅是一个原始的、未解释的内存区域，用于存储数据。
存储对象通常不直接由用户操作，而是作为张量（Tensor）的底层数据结构。每个张量都有一个与之关联的存储对象，张量操作实际上是对存储对象中的数据进行操作。
import torch
# 创建一个 PyTorch 张量
tensor = torch.tensor([1, 2, 3])
# 创建一个 PyTorch 存储对象（Storage）
storage = tensor.storage()
# 使用 torch.is_storage() 检查 tensor 是否是存储对象
is_storage_tensor = torch.is_storage(tensor)
is_storage_storage = torch.is_storage(storage)
print(f"Is 'tensor' a storage: {is_storage_tensor}") # 应该输出 False
print(f"Is 'storage' a storage: {is_storage_storage}") # 应该输出 True