How can I make it, because y tensor are labels of segmentation (instance segmentation) ?
For the input to this function it has 1-dim as follow : tensor([ 8.4146,  6.0662,  6.8091,  8.1898, 10.7489, 10.3809], device='cuda:0',        grad_fn=<SubBackward0>)