我训练了一个可以在桌面浏览器（Mac上为Chrome＆Safari）上正常运行的模型，但是在iPhone上的Safari上运行失败（我得到了NaN）。
我已经确定oneHot的基本版本在两个平台上都有差异：
例如
tf.oneHot（[1]，5）
创建：
我的桌面浏览器上的[0,1,0,0,0]（按预期）
在iPhone / Safari上为[0,0,0,0,0]
我创建了一个演示页面，该页面显示了https://beta.observablehq.com/@elaval/issue-with-tensorflowjs-on-iphone-safari上的这种行为
（请尝试一下，告诉我在不同的平台（例如Android）上获得的是[0,1,0,0,0]还是[0,0,0,0,0]。
对此有什么根本原因的任何想法？这在所有手机上都失败还是仅在Safari中失败？我们如何克服这个问题？
谢谢