I have a matrix X with N rows and D columns, and I performs several operations on X, like gathering several rows using tf.gather and s
X
N
D