Función torch.as_tensor de PyTorch
Manual de referencia de Pytorch torch
torch.as_tensorEs la función en PyTorch para convertir datos en tensores. Contorch.tensordiferente, comparte memoria de datos tanto como sea posible.
Definición de la función
torch.as_tensor(data, dtype=None, device=None)
Ejemplo de uso
Ejemplo
import torch
import numpy as np
# Crear a partir de un array de NumPy (memoria compartida)
numpy_array = np.array([1, 2, 3])
x = torch.as_tensor(numpy_array)
# Modificar el array de NumPy
numpy_array[0] = 100
print(Tensor:, x) # El tensor también cambiará
import numpy as np
# Crear a partir de un array de NumPy (memoria compartida)
numpy_array = np.array([1, 2, 3])
x = torch.as_tensor(numpy_array)
# Modificar el array de NumPy
numpy_array[0] = 100
print(Tensor:, x) # El tensor también cambiará
El resultado de salida es:
张量: tensor([100, 2, 3])
Otras extensiones