View source on GitHub |
Create a numpy ndarray from a tensor.
tf.make_ndarray( tensor )
Create a numpy ndarray with the same shape and data as the tensor.
# Tensor a has shape (2,3) a = tf.constant([[1,2,3],[4,5,6]]) proto_tensor = tf.make_tensor_proto(a) # convert `tensor a` to a proto tensor tf.make_ndarray(proto_tensor) # output: array([[1, 2, 3], # [4, 5, 6]], dtype=int32) # output has shape (2,3)
Args | |
---|---|
tensor | A TensorProto. |
Returns | |
---|---|
A numpy array with the tensor contents. |
Raises | |
---|---|
TypeError | if tensor has unsupported type. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/make_ndarray