garage.torch.utils module¶

Utility functions for PyTorch.

np_to_torch(array_dict)[source]¶

Convert numpy arrays to PyTorch tensors.

Args:

dict (dict): Dictionary of data in numpy arrays.

Returns:	Dictionary of data in PyTorch tensors.

torch_to_np(value_in)[source]¶

Convert PyTorch tensors to numpy arrays.

Args:

value_in (tuple): Tuple of data in PyTorch tensors.

Returns:	Tuple of data in numpy arrays.