Error in loss function

[Max-Well-Wang]

Dear author, when the batchsize is greater than one, the generated boundary length of each batchsize is inconsistent, which leads to different dimensions for loss. Can you tell me how to solve this problem?