Skip to content

Instantly share code, notes, and snippets.

@thierryherrmann
Created August 3, 2020 00:05
Show Gist options
  • Save thierryherrmann/82c983caff614f1330bfb2fffa77a8a7 to your computer and use it in GitHub Desktop.
Save thierryherrmann/82c983caff614f1330bfb2fffa77a8a7 to your computer and use it in GitHub Desktop.
input_X = graph.get_tensor_by_name('my_train_X:0')
input_y = graph.get_tensor_by_name('my_train_y:0')
output_1 = graph.get_tensor_by_name('StatefulPartitionedCall_1:0')
output_2 = graph.get_tensor_by_name('StatefulPartitionedCall_1:1')
out_val_1, out_val_2 = session.run([output_1, output_2],
feed_dict={input_X: X_train[0:1], input_y: y_train[0:1]})
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment