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'