I am facing a problem to implement a Sparse Matrix support update for a custom optimizer in TensorFlow v2.x
# Dense Matrix