Could you please provide the keras implement？

[Ianwtg]

Hi， thank you for your great work.
Could you please provide the keras implement？

[Alex Stergiou]

Hi @Ianwtg ,

We have not created a Keras/TensorFlow implementation for the project. You could however implement it with Keras/TensorFlow build-in functions similar to the PyTorch CPU version seen in lines 207-210.

Best,
Alex