You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
6 lines
144 B
Python
6 lines
144 B
Python
2 years ago
|
import tensorflow as tf
|
||
|
|
|
||
|
|
def my_loss_fn(y_true, y_pred):
|
||
|
|
|
||
|
|
return tf.reduce_mean(squared_difference, axis=-1) # Note the `axis=-1`
|