Creating a custom LSTMCell in Tensorflow with GPU support

Question

I would like to integrate an attentional component into the LSTM model I\'m creating. Unfortunately, with tensorflow 2.3.1 that I\'m using, it appears that if you subclass