torch.is_storage

torch.is_storage(obj, /)

如果 obj 是一个 PyTorch storage 对象，则返回 True。

参数

obj (对象) – 要测试的对象

返回类型

typing_extensions.TypeIs[Union[TypedStorage, UntypedStorage]]