jax.experimental.multihost_utils.assert_equal

jax.experimental.multihost_utils.assert_equal#

jax.experimental.multihost_utils.assert_equal(in_tree, fail_message='')[原始碼]#

驗證所有主機是否具有相同的值樹狀結構。