1:1 TOR103 Import of deprecated function torch.utils._pytree._register_pytree_node
3:1 TOR101 Use of deprecated function torch.utils._pytree._register_pytree_node
6:1 TOR101 Use of deprecated function torch.utils._pytree._register_pytree_node
9:1 TOR101 Use of deprecated function torch.utils._pytree._register_pytree_node