Skip to content

Instantly share code, notes, and snippets.

@miquelbeltran

miquelbeltran/input_keras.kt

Created September 7, 2018 11:40
Show Gist options
  • Save miquelbeltran/d23ebcb0f3e842a86fb4b5421b50fb03 to your computer and use it in GitHub Desktop.
Save miquelbeltran/d23ebcb0f3e842a86fb4b5421b50fb03 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
input_keras.kt
// Input is a 1x784 Tensor
val inputDims = intArrayOf(1, 784)
// Output is a 1x10 Tensor
val outputDims = intArrayOf(1, 10)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment