Torch argmax()在Mac MPS上返回错误结果，CPU上结果正常

• 存在一个名为target_mixtures的张量矩阵，执行print(target_mixtures[0])得到输出：

tensor([0.0010, 0.6827, 0.0010, 0.0261, 0.0010, 0.0010, 0.0010, 0.0010, 0.0010, 0.0010, 0.0010, 0.0010, 0.0010, 0.0010, 0.2744, 0.0010, 0.0010, 0.0010, 0.0010, 0.0010], device='mps:0')

• 将张量转移到CPU执行print(torch.argmax(target_mixtures[0].cpu()))，返回预期结果1（对应张量中的最大值0.6827）；
• 但直接在Mac MPS设备上执行print(torch.argmax(target_mixtures[0]))时，返回结果为0，且对所有target_mixtures[i]调用torch.argmax（MPS设备上）均返回0。

内容的提问来源于stack exchange，提问作者kldivergence