torch.is_storage

torch.is_storage(obj, /)

如果 obj 是 PyTorch 存储对象，则返回 True。

参数

obj (Object) – 要测试的对象

返回类型

typing_extensions.TypeIs[Union[TypedStorage, UntypedStorage]]