torch.is_storage

torch.is_storage(obj, /)

如果 obj 是一個 PyTorch storage 物件，則返回 True。

引數

obj (Object) – 要測試的物件

返回型別

TypeIs[Union[TypedStorage, UntypedStorage]]