All Questions
2 questions
0
votes
1
answer
1k
views
Error Saving & Loading Tensorflow/Keras Model With Custom Classes/Functions
I recently created a Tensorflow/Keras model with Keras Transformers. To do this, the custom PositionalEmbedding & TransformerEncoder classes were created and used to build the model architecture. ...
- 386
0
votes
1
answer
749
views
Custom function with multiple argument and one return value in map_fn for tensor object in Tensorflow
I have two tensors t1 and t2 (shape=(64,64,3), dtype=tf.float64). I want to execute a custom function "func" which takes two tensors as input and returns one new tensor.
@tf.function
def ...
- 327