Hello, I have a neural network with two input tensors:
a shape (batch_size=128, 3)
b shape (60, batch_size=128, 1)

I need to augment a to have the same shape as b:
a.repeat(60, 1, 1) -> a shape (60, 128, 3)
Then I want to put a and b in the same neural network that accepts two dimensions. I need to reshape so that the 60 can be considered the same as the batch size:
result = neural_network(a.reshape(-1, 3), b.reshape(-1, 1))
This result now has the same ((60 x 128), 1)
Now I want to do the reshape so that the result has the shape (60, 128, 1) and it has the same order than a and b had initially. If I do:
result_reshaped = result.reshape(60, 128, 1) I will get that, however, how can I be sure that the order is correct?
Thanks!