[ONNX] How to gather on a tensor with two-dim indexing?

1

Hi,

How can I perform the following without getting a Gather node in my onnx graph? As the Gather node gives me an error in TensorRT 7.

x = data[:, x_indices, y_indices]

data is tensor of size[32, 64, 1024]
x_indices is tensor of size [50000,] → range of indices 0 to 31
y_indices is tensor of size [50000,] → range of indices 0 to 1023

The size of tensor x would be [32,50000]

Thanks in advance.