torch.is_storage

torch.is_storage(obj, /)

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

参数

obj (Object) – 要测试的对象

返回类型

TypeIs[Union[TypedStorage, UntypedStorage]]