search

sascha-kirch / DeepSaki

DeepSaki is an add-on to TensorFlow. It provides a variaty of custom classes ranging from activation functions to entire models, helper functions to facilitate connectiong to your, compute HW and many more!
https://sascha-kirch.github.io/DeepSaki/
MIT License
10 stars 0 forks source link

add tf.keras.layers.InputSpec to custom layers layers #13

Open sascha-kirch opened 1 year ago

sascha-kirch commented 1 year ago

with tf.keras.layers.InputSpec(dim=xxx) one can specify the specification for an input when the layer is called.

Dont forget: unit test

sascha-kirch commented 1 year ago

https://www.tensorflow.org/api_docs/python/tf/keras/layers/InputSpec. I could also add dtypey, especially for the Fourier layers that expect complex inputs.