shelton-xiaoshuaichen
Results
2
comments of
shelton-xiaoshuaichen
I have the same question
should fix to follow? def index_matrix_to_pairs(index_matrix): # [[3,1,2], [2,3,1]] -> [[[0, 3], [1, 1], [2, 2]], # [[0, 2], [1, 3], [2, 1]]] replicated_first_indices = tf.range(tf.shape(index_matrix)[-1]) #[0, 1, 2] rank...