jax.tree_util.keystr

jax.tree_util.keystr(keys)

帮助程序，用于美化打印键的元组。

参数:

keys (KeyPath) – 一个 KeyEntry 元组或任何可以转换为字符串的类。

返回:

一个连接所有键的字符串表示的字符串。

示例

>>> import jax
>>> keys = (0, 1, 'a', 'b')
>>> jax.tree_util.keystr(keys)
'01ab'