Меня немного озадачивает следующее поведение jax (версия 0.6.2).
Обычно, если я транслирую дерево префиксов в структурное дерево, значение префикса транслируется в структуру:
Код: Выделить всё
out_tree = jax.tree.broadcast(True, (False, False), is_leaf=lambda x: x is None) # Outputs (True, True)
Однако, если дерево структуры содержит None (отмечено как листья), тогда в широковещательном сообщении вместо этого будут передаваться значения None:
Код: Выделить всё
out_tree = jax.tree.broadcast(True, (None, False), is_leaf=lambda x: x is None) # Outputs (None, True)
Что касается трансляции, я ожидал, что дерево структуры будет именно структурой, и что значения его листьев не должны влиять на трансляцию. Какова цель такого поведения?
