2021-02-26 22:54:13.146272: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 2989 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1050, pci bus id: 0000:01:00.0, compute capability: 6.1)
2021-02-26 22:54:13.146839: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
Traceback (most recent call last):
  File "E:/att.py", line 61, in <module>
    m = model_attention_applied_after_lstm()
  File "E:/att.py", line 35, in model_attention_applied_after_lstm
    attention_mul = attention_3d_block(lstm_out)
  File "E:/att.py", line 27, in attention_3d_block
    output_attention_mul = merge([inputs, a_probs], name='attention_mul', mode='mul')
TypeError: 'module' object is not callable