def METHOD_NAME(self, tensor_list: List[StatefulTensor]) -> None:
    assert self._stateful_tensor_list == [], "Can't register stateful tensors for manager twice"
    self._stateful_tensor_list = tensor_list
    for t in self._stateful_tensor_list:
        assert isinstance(t, StatefulTensor)
        t.trans_state = types.MethodType(functools.partial(self._trans_state, t.trans_state), t)