nip.utils.data.tensordict_to_numpy_dict

nip.utils.data.tensordict_to_numpy_dict#

nip.utils.data.tensordict_to_numpy_dict(data: TensorDictBase) dict[source]#

Convert a TensorDict to a dict of numpy arrays.

Parameters:

data (TensorDictBase) – The TensorDict to convert.

Returns:

data (dict) – The dict of numpy arrays.