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'