How to access tensor_content values in TensorProto in TensorFlow?

Decode tensor_array bytes and then reshape with given shape:

for node in tf.get_default_graph.as_graph_def().node:
    tensor_bytes = node.attr["value"].tensor.tensor_content
    tensor_dtype = node.attr["value"].tensor.dtype
    tensor_shape = [x.size for x in node.attr["value"].tensor.tensor_shape.dim]
    tensor_array = tf.decode_raw(tensor_bytes, tensor_dtype)
    tensor_array = tf.reshape(tensor_array, tensor_shape)

from tensorflow.python.framework import tensor_util

for n in tf.get_default_graph().as_graph_def().node:
    print tensor_util.MakeNdarray(n.attr['value'].tensor)

Tags:

Tensorflow