Skip to content

Instantly share code, notes, and snippets.

@alstat
Last active October 27, 2019 15:57
Show Gist options
  • Save alstat/7ade32577acd6c5f22007eaa6e3ac894 to your computer and use it in GitHub Desktop.
Save alstat/7ade32577acd6c5f22007eaa6e3ac894 to your computer and use it in GitHub Desktop.
layer = Dense(4, 2, tf.nn.relu)
layer(data.xtrn[1:2, :])
#> tf.Tensor([[12.937485 0. ]], shape=(1, 2), dtype=float32)
layer(data.xtrn[1:5, :])
#> tf.Tensor(
#> [[12.937484 0. ]
#> [12.557415 0. ]
#> [13.761768 0. ]
#> [14.996015 0. ]], shape=(4, 2), dtype=float32)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment