ejemplo de código de matriz numpy a tensor pytorch

Ejemplo 1: convertir numpy en antorcha

torch.from_numpy(your_array)

Ejemplo 2: tensot a numpy pytorch

#Back and forth between torch tensor and numpy

#np --> tensot
torch.from_numpy(your_numpy_array)

#tensor --> np
your_torch_tensor.numpy()

Ejemplo 3: convertir el tensor de pytorch en numpy

na = a.to('cpu').numpy()

---

---

Utiliza Nuestro Buscador