probly.representation.sample.torch_axis_tracking

Axis-tracking for PyTorch tensors.