The output seems to have the right shape but when I pass it to the binary_loss_fn function it expects 8 elements (o1, o2, o3, o4, o5, o6, o7, o8 = outputs) but finds 2 because the first dimension is referred to the batch size. However, can I use an array as a parameter on pos_weight using only a loss function? I hope I have understood correctly