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'