TensorFlow

API

 tf / make_ndarray


Create a numpy ndarray from a tensor.

Create a numpy ndarray with the same shape and data as the tensor.

For example:

# 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)

tensor A TensorProto.

A numpy array with the tensor contents.

TypeError if tensor has unsupported type.

此页内容是否对您有帮助