How can you specify a different matrix permutation for each element in a batch?

Question

I have a constant symmetric matrix A with shape (50,50) and inputs x with shape (batch_size, 50) where each entry is an i