Hi,

Is there a way in PyTorch to input a network (let’s assume a basic MLP network) a 3-dimensional tensor of shape [batch_size, x, y] and get at the output a 2-dimensional tensor of shape [batch_size, x+y]? I know I can reshape the input to [batch_size, x*y] and then the solution will be simple, but I assume x and y to be high dimensional and I want to avoid this multiplication.

Thanks!