kaira.data.datasets.UniformDataset

Inheritance diagram for UniformDataset

class kaira.data.datasets.UniformDataset(length: int, shape: int | Tuple[int, ...] = (128,), low: float = 0.0, high: float = 1.0, seed: int | None = None)

Bases: Dataset

Dataset for uniformly distributed tensor data.

Generates tensors with uniformly distributed random values on-demand. Useful for noise generation and random signal experiments.

Methods

__init__

Initialize the uniform dataset.

__init__(length: int, shape: int | Tuple[int, ...] = (128,), low: float = 0.0, high: float = 1.0, seed: int | None = None)

Initialize the uniform dataset.

Parameters:

• length – Number of samples in the dataset

• shape – Shape of each tensor (int for 1D, tuple for multi-dimensional)

• low – Lower bound for uniform distribution

• high – Upper bound for uniform distribution

• seed – Random seed for reproducibility