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'