About batch size

[silencecmsj]

When I set batch_size to 2, the output of the encoder is abnormal, as shown below:

This will lead to problems in the final result, how to solve it?

B, N, _, _ = input.shape  # type: int, int, int, int

ValueError: too many values to unpack (expected 4)