pred is (52736,32000) and gold (52736,1). Can i divide the tensor to have different calculation for each GPU , like i have 8 GPUs and i think the problem is that the calculation is acculamated. So if i divide the tensor to 8 parts and do the calculation seperately and then add up , can this work ?