Skip to content

Instantly share code, notes, and snippets.

@zaidalyafeai
Last active April 3, 2018 23:03
Show Gist options
  • Save zaidalyafeai/461d61fbe96e9db68ef2ecdfcb22b8ef to your computer and use it in GitHub Desktop.
Save zaidalyafeai/461d61fbe96e9db68ef2ecdfcb22b8ef to your computer and use it in GitHub Desktop.
const convlayer = tf.layers.conv2d({
inputShape: [28, 28, 1],
kernelSize: 5,
filters: 8,
strides: 1,
activation: 'relu',
kernelInitializer: 'VarianceScaling'
});
const input = tf.zeros([1,28,28,1]);
const output = convlayer.apply(input);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment