Skip to content

Instantly share code, notes, and snippets.

@fabclmnt

fabclmnt/tfFunction.py

Created May 7, 2020 14:36
Show Gist options
  • Save fabclmnt/2e0dcabb419484c197466e8e79c7e973 to your computer and use it in GitHub Desktop.
Save fabclmnt/2e0dcabb419484c197466e8e79c7e973 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
tfFunction.py
import tensorflow as tf
# Adding the decorator
@tf.function
def matprod(x, y):
return tf.matmul(x, y)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment