Question

我正在尝试在Compute引擎和Google Cloud TPU上运行自定义估算器，但是出现错误：

ValueError: shape [2] cannot be sharded 8 ways along dimension 0

我不知道是什么原因或如何解决它-知道吗？

安迪（Andy P。）

Answer 1

您似乎正在使用一个二值向量。理想情况下，TPU的所有尺寸都可以被8整除。如果您只是在玩耍，我建议您制作8个长矢量。