tf2.py

import numpy as np
import tensorflow as tf
 
np.random.seed(42)
 
x = tf.constant(np.random.rand(3,3))
y = tf.constant(np.random.rand(3,3))

@tf.function
def matmul(x, y):
    return tf.matmul(x, y)
 
print(matmul(x, y).numpy())