torch.is_storage

torch.is_storage(obj, /)

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

引數

obj (物件) – 要測試的物件

返回型別

typing_extensions.TypeIs[Union[TypedStorage, UntypedStorage]]