Функция PyTorch torch.is_storage
Справочное руководство Pytorch torch
torch.is_storageЭто функция в PyTorch, используемая для проверки, является ли объект объектом storage.
Определение функции
torch.is_storage(obj)
Пример использования
Пример
import torch
# Создать тензор
x = torch.tensor([1, 2, 3])
# Проверить, является ли это объектом storage
print(torch.is_storage(x))
print(torch.is_storage(x.storage()))
# Создать тензор
x = torch.tensor([1, 2, 3])
# Проверить, является ли это объектом storage
print(torch.is_storage(x))
print(torch.is_storage(x.storage()))
Результат вывода:
False True
Другие расширения