pytupli.dataset.TorchTupleParser

class TorchTupleParser[source]

Bases: BaseTupleParser

Parser for converting lists of observations, actions, rewards, terminal flags, and timeout flags into PyTorch tensors.

Methods

parse_lists

Converts lists to PyTorch tensors.

classmethod parse_lists(obs: list[list[float]], act: list[list[float]], rew: list[float], term: list[bool], trunc: list[bool]) tuple[source]

Converts lists to PyTorch tensors.