Answers for "ndarray to tensor"

3

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)
Posted by: Guest on January-14-2021
0

how do i turn a tensor into a numpy array

import torch

# Create PyTorch tensor
A_torch = torch.tensor([1, 2])

# Convert tensor to NumPy array
A_np = A_torch.numpy()
Posted by: Guest on January-14-2021

Python Answers by Framework

Browse Popular Code Answers by Language