traja.accessor.TrajaAccessor.to_tensor

TrajaAccessor.to_tensor(columns: list | None = None)

Convert trajectory to PyTorch tensor (if torch is available).

Parameters:

columns – List of columns to include (default: [‘x’, ‘y’] or [‘x’, ‘y’, ‘z’])

Returns:

Trajectory as tensor, or numpy array if torch not available

Return type:

torch.Tensor

Note

Requires PyTorch. Install with: pip install torch