probly.representation.sample.torch_functions.TorchSampleInternals¶

class probly.representation.sample.torch_functions.TorchSampleInternals(create: TorchSampleCreator[D], tensor: D, sample_dim: int, weights: torch.Tensor | None = None)[source]¶

Bases: Generic

Internal information about a sample tensor.

create: TorchSampleCreator¶

sample_dim: int¶

tensor: D¶

weights: Tensor | None¶