Thank you for your time, could you just explain me more, about this:
 label=torch.tensor([[0, 0, 0], [0, 1, 0], [0, 0, 2]])
		output=torch.randn(3, 3, requires_grad=True)
		loss=criterion(output, label.float())