PyTorch torch.is_storage Función

Pytorch torch 参考手册Manual de referencia de PyTorch torch

torch.is_storageEs una función en PyTorch utilizada para comprobar si un objeto es un objeto de almacenamiento.

Definición de la función

torch.is_storage(obj)

Ejemplo de uso

Ejemplo

import torch

# Crear tensor
x = torch.tensor([1, 2, 3])

# Comprobar si es un objeto de almacenamiento
print(torch.is_storage(x))
print(torch.is_storage(x.storage()))

El resultado de salida es:

False
True

Pytorch torch 参考手册Manual de referencia de PyTorch torch

Otras extensiones