Skip to content

Instantly share code, notes, and snippets.

@YusukeSuzuki
Created November 9, 2018 08:22
Show Gist options
  • Select an option

  • Save YusukeSuzuki/6db4041a70742a2f089590142c2a8320 to your computer and use it in GitHub Desktop.

Select an option

Save YusukeSuzuki/6db4041a70742a2f089590142c2a8320 to your computer and use it in GitHub Desktop.
import tensorflow as tf
import numpy as np
from keras.layers import Conv2D
img = tf.constant(np.ones([16,128,128,3]), dtype=tf.float32)
conv = Conv2D(32, 3)
y = conv(img)
graph = tf.get_default_graph()
print(graph.as_graph_def())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment