convert pytorch tensor to numpy array code example

Example 1: tensot to numpy pytorch

#Back and forth between torch tensor and numpy

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

#tensor --> np
your_torch_tensor.numpy()

Example 2: convert pytorch tensor to numpy

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

Example 3: convert torch to numpy

your_tensor.numpy()

Example 4: convert tensor to numpy array

import tensorflow as tf

a = tf.constant([[1, 2], [3, 4]])                 
b = tf.add(a, 1)

a.numpy()
# array([[1, 2],
#        [3, 4]], dtype=int32)

b.numpy()
# array([[2, 3],
#        [4, 5]], dtype=int32)

tf.multiply(a, b).numpy()
# array([[ 2,  6],
#        [12, 20]], dtype=int32)

Tags:

Misc Example