NumPy和Tensor之间得转换

Converting between a TensorFlow tf.Tensors and a NumPy ndarray is easy:

1.TensorFlow operations automatically convert NumPy ndarrays to Tensors.
2,NumPy operations automatically convert Tensors to NumPy ndarrays.

Tensors are explicitly converted to NumPy ndarrays using their .numpy() method. These conversions are typically cheap since the array and tf.Tensor share the underlying memory representation, if possible. However, sharing the underlying representation isn’t always possible since the tf.Tensor may be hosted in GPU memory while NumPy arrays are always backed by host memory, and the conversion involves a copy from GPU to host memory.

import numpy as np

ndarray = np.ones([3, 3])

print("TensorFlow operations convert numpy arrays to Tensors automatically")
tensor = tf.multiply(ndarray, 42)
print(tensor)


print("And NumPy operations convert Tensors to numpy arrays automatically")
print(np.add(tensor, 1))

print("The .numpy() method explicitly converts a Tensor to a numpy array")
print(tensor.numpy())
上一篇:DEEP LEARNING WITH PYTORCH: A 60 MINUTE BLITZ | TENSORS


下一篇:pytorch datasets与dataloader阐释说明