因为Pytorch对Apple M1芯片的支持的问题，此处的代码直接在使用m1芯片的mac上是无法直接运行的。原因在于torch.argmax()这个函数，运行时会产生一个“-9xxxxxxx”的错误值，可替代的改法是使用torch.max（）函数，例如“_，y=y.max（地面=1）”。这个问题在全书多处地方出现，所有使用了argmax（）函数的地方都有可能出现。是不是应该指出一下？
因为Pytorch对Apple M1芯片的支持的问题，此处的代码直接在使用m1芯片的mac上是无法直接运行的。原因在于torch.argmax()这个函数，运行时会产生一个“-9xxxxxxx”的错误值，可替代的改法是使用torch.max（）函数，例如“_，y=y.max（地面=1）”。这个问题在全书多处地方出现，所有使用了argmax（）函数的地方都有可能出现。是不是应该指出一下？