Skip to content

Instantly share code, notes, and snippets.

@chaserileyroberts

chaserileyroberts/tn_model.py

Created February 7, 2020 20:05
Show Gist options
  • Save chaserileyroberts/f8027729a0c01567c62c52dbbf25206f to your computer and use it in GitHub Desktop.
Save chaserileyroberts/f8027729a0c01567c62c52dbbf25206f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
tn_model.py
tn_model = tf.keras.Sequential(
[
tf.keras.Input(shape=(2,)),
Dense(1024, activation=tf.nn.relu),
# Here use a TN layer instead of the dense layer.
TNLayer(),
Dense(1, activation=None)
]
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment