G.M:

.long()


Thank you for your reply, @G.M. That solved the problem.
It seems there is another problem about targets = torch.cat((targets, tragets_zero)):
RuntimeError:  Tensors must have same number of dimensions: got 2 and 1 at /opt/conda/conda-bld/pytorch_.....