Using batched input with tf.math.invert_permutation

Question

I have a tensor holding a batch of permutations of the integers 0 to time-1 which e.g. has the shape

[batch,time]

N