Skip to content

Instantly share code, notes, and snippets.

@alseambusher

alseambusher/conv2d.usage.ts

Created January 23, 2018 22:24
Show Gist options
  • Save alseambusher/87d06f98c054eafd9d9f36438c12d74e to your computer and use it in GitHub Desktop.
Save alseambusher/87d06f98c054eafd9d9f36438c12d74e to your computer and use it in GitHub Desktop.
Download ZIP
conv2d usage
Raw
conv2d.usage.ts
// 32 5x5 filters
var network = conv2d(tensor, 32, 5, graph);
// 32 5x5 filters, stride 2, "same" padding with relu activation
var network = conv2d(tensor, 32, 5, graph, 2, "SAME", undefined, (layer, graph) => {return graph.relu(layer)});
// applying some kernel_initializer
var network = conv2d(x, 32, 5, g, undefined, undefined, undefined, undefined, new RandomUniformInitializer(0, 0.5));
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment