https://www.tensorflow.org/ranking/api_docs/python/tfr/keras/layers/DocumentInteractionAttention?authuser=4
tfr.keras.layers.DocumentInteractionAttention | TensorFlow Ranking
Cross Document Interaction Attention layer.
tfrkeraslayerstensorflowranking