jax.experimental.multihost_utils.assert_equal

jax.experimental.multihost_utils.assert_equal#

jax.experimental.multihost_utils.assert_equal(in_tree, fail_message='')[源代码]#

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