jax.experimental.multihost_utils.assert_equal

jax.experimental.multihost_utils.assert_equal#

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

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