jax.experimental.multihost_utils.assert_equal

jax.experimental.multihost_utils.assert_equal(in_tree, fail_message='')

验证所有主机是否具有相同的树形值。

参数：

fail_message (str)