{
  "$schema": "https://ambiqai.github.io/helia-ui/schema/reference-model-1.json",
  "generatedFrom": {
    "sourceCommit": "f341fb11f7f5d77a4974ba8273c4cd55d67c21f0",
    "tool": "pyref",
    "version": "1.7.3"
  },
  "language": "python",
  "modules": [
    {
      "description": "Unified public-Keras hooks for deterministic preprocessing and augmentation.",
      "name": "base_augmentation",
      "path": "helia_edge.layers.preprocessing.base_augmentation",
      "submodules": [],
      "summary": "Unified public-Keras hooks for deterministic preprocessing and augmentation.",
      "symbols": [
        {
          "description": "Shared construction-time configuration; never validates live tensors.",
          "examples": [],
          "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentationParams",
          "kind": "class",
          "language": "python",
          "members": [
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentationParams.model_config",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "model_config",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "model_config = ConfigDict(extra='forbid', frozen=True)",
              "source": {
                "line": 12,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L12"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentationParams.seed",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "seed",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "seed: int | None = None",
              "source": {
                "line": 13,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L13"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentationParams.auto_vectorize",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "auto_vectorize",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "auto_vectorize: bool = False",
              "source": {
                "line": 14,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L14"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentationParams.data_format",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "data_format",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "data_format: Literal['channels_first', 'channels_last'] = 'channels_last'",
              "source": {
                "line": 15,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L15"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentationParams.device",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "device",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "device: str = 'cpu'",
              "source": {
                "line": 16,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L16"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentationParams.aligned_targets",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "aligned_targets",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "aligned_targets: tuple[str, ...] | None = None",
              "source": {
                "line": 17,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L17"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentationParams.aligned_masks",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "aligned_masks",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "aligned_masks: tuple[str, ...] | None = None",
              "source": {
                "line": 18,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L18"
              },
              "summary": ""
            }
          ],
          "name": "BaseAugmentationParams",
          "params": [],
          "raises": [],
          "returns": [],
          "signature": "BaseAugmentationParams()",
          "source": {
            "line": 9,
            "path": "helia_edge/layers/preprocessing/base_augmentation.py",
            "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L9"
          },
          "summary": "Shared construction-time configuration; never validates live tensors."
        },
        {
          "description": "Base for rank-specific transforms, with one sampling/application contract.\n\nOverride ``augment_samples`` or ``augment_sample`` and optionally\n``get_random_transformations``. Sample parameters once per signal batch;\njoint transforms reuse them across aligned signals, targets and masks.\nPer-example parameter leaves must have a leading batch axis. Deterministic\nsubclasses set ``training_only=False``. Bases are not serialized transforms.",
          "examples": [],
          "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation",
          "kind": "class",
          "language": "python",
          "members": [
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.SAMPLES",
              "kind": "constant",
              "language": "python",
              "members": [],
              "name": "SAMPLES",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "SAMPLES = 'data'",
              "source": {
                "line": 31,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L31"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.LABELS",
              "kind": "constant",
              "language": "python",
              "members": [],
              "name": "LABELS",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "LABELS = 'labels'",
              "source": {
                "line": 32,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L32"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.TARGETS",
              "kind": "constant",
              "language": "python",
              "members": [],
              "name": "TARGETS",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "TARGETS = 'targets'",
              "source": {
                "line": 33,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L33"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.TRANSFORMS",
              "kind": "constant",
              "language": "python",
              "members": [],
              "name": "TRANSFORMS",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "TRANSFORMS = 'transforms'",
              "source": {
                "line": 34,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L34"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.NDIMS",
              "kind": "constant",
              "language": "python",
              "members": [],
              "name": "NDIMS",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "NDIMS = 4",
              "source": {
                "line": 35,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L35"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.training_only",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "training_only",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "training_only = True",
              "source": {
                "line": 36,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L36"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.joint",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "joint",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "joint = False",
              "source": {
                "line": 37,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L37"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.seed",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "seed",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "seed = params.seed",
              "source": {
                "line": 59,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L59"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.auto_vectorize",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "auto_vectorize",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "auto_vectorize = params.auto_vectorize",
              "source": {
                "line": 60,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L60"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.data_format",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "data_format",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "data_format = params.data_format",
              "source": {
                "line": 61,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L61"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.device",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "device",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "device = params.device",
              "source": {
                "line": 62,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L62"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.aligned_targets",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "aligned_targets",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "aligned_targets = params.aligned_targets",
              "source": {
                "line": 63,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L63"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.aligned_masks",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "aligned_masks",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "aligned_masks = params.aligned_masks",
              "source": {
                "line": 64,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L64"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.generator",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "generator",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "generator = keras.random.SeedGenerator(self.seed)",
              "source": {
                "line": 65,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L65"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.random_generator",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "random_generator",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "random_generator",
              "source": {
                "line": 68,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L68"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.ch_axis",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "ch_axis",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "ch_axis",
              "source": {
                "line": 72,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L72"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.data_axis",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "data_axis",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "data_axis",
              "source": {
                "line": 76,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L76"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.height_axis",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "height_axis",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "height_axis",
              "source": {
                "line": 80,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L80"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.width_axis",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "width_axis",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "width_axis",
              "source": {
                "line": 84,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L84"
              },
              "summary": ""
            },
            {
              "description": "Return batched parameter tensors, or None for deterministic transforms.",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.get_random_transformations",
              "kind": "method",
              "language": "python",
              "members": [],
              "name": "get_random_transformations",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "get_random_transformations(input_shape)",
              "source": {
                "line": 92,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L92"
              },
              "summary": "Return batched parameter tensors, or None for deterministic transforms."
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_sample",
              "kind": "method",
              "language": "python",
              "members": [],
              "name": "augment_sample",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "augment_sample(inputs)",
              "source": {
                "line": 96,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L96"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_samples",
              "kind": "method",
              "language": "python",
              "members": [],
              "name": "augment_samples",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "augment_samples(inputs)",
              "source": {
                "line": 99,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L99"
              },
              "summary": ""
            },
            {
              "description": "Apply geometry to selected targets using the signal parameters.",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_targets",
              "kind": "method",
              "language": "python",
              "members": [],
              "name": "augment_targets",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "augment_targets(inputs)",
              "source": {
                "line": 106,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L106"
              },
              "summary": "Apply geometry to selected targets using the signal parameters."
            },
            {
              "description": "Apply geometry to selected masks using the signal parameters.",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_masks",
              "kind": "method",
              "language": "python",
              "members": [],
              "name": "augment_masks",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "augment_masks(inputs)",
              "source": {
                "line": 110,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L110"
              },
              "summary": "Apply geometry to selected masks using the signal parameters."
            },
            {
              "description": "Apply to a tensor or schema, with optional pre-sampled parameters.",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.batch_augment",
              "kind": "method",
              "language": "python",
              "members": [],
              "name": "batch_augment",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "batch_augment(inputs, transformations=None)",
              "source": {
                "line": 138,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L138"
              },
              "summary": "Apply to a tensor or schema, with optional pre-sampled parameters."
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.call",
              "kind": "method",
              "language": "python",
              "members": [],
              "name": "call",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "call(inputs, training=None, transformations=None)",
              "source": {
                "line": 215,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L215"
              },
              "summary": ""
            },
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.get_config",
              "kind": "method",
              "language": "python",
              "members": [],
              "name": "get_config",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "get_config()",
              "source": {
                "line": 229,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L229"
              },
              "summary": ""
            }
          ],
          "name": "BaseAugmentation",
          "params": [],
          "raises": [],
          "returns": [],
          "signature": "BaseAugmentation(\n    seed=None,\n    auto_vectorize=False,\n    data_format=None,\n    device='cpu',\n    aligned_targets=None,\n    aligned_masks=None,\n    **kwargs={},\n)",
          "source": {
            "line": 21,
            "path": "helia_edge/layers/preprocessing/base_augmentation.py",
            "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L21"
          },
          "summary": "Base for rank-specific transforms, with one sampling/application contract."
        },
        {
          "description": "One-dimensional signals with optional batch axis.\n\n\n\nBase class: [BaseAugmentation](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation).\n\nInherited from [BaseAugmentation](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation): [augment_masks()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_masks), [augment_sample()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_sample), [augment_samples()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_samples), [augment_targets()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_targets), [batch_augment()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.batch_augment), [call()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.call), [get_config()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.get_config), [get_random_transformations()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.get_random_transformations).",
          "examples": [],
          "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation1D",
          "kind": "class",
          "language": "python",
          "members": [
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation1D.NDIMS",
              "kind": "constant",
              "language": "python",
              "members": [],
              "name": "NDIMS",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "NDIMS = 3",
              "source": {
                "line": 244,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L244"
              },
              "summary": ""
            }
          ],
          "name": "BaseAugmentation1D",
          "params": [],
          "raises": [],
          "returns": [],
          "signature": "BaseAugmentation1D(\n    seed=None,\n    auto_vectorize=False,\n    data_format=None,\n    device='cpu',\n    aligned_targets=None,\n    aligned_masks=None,\n    **kwargs={},\n)",
          "source": {
            "line": 241,
            "path": "helia_edge/layers/preprocessing/base_augmentation.py",
            "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L241"
          },
          "summary": "One-dimensional signals with optional batch axis."
        },
        {
          "description": "Two-dimensional images with optional batch axis.\n\n\n\nBase class: [BaseAugmentation](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation).\n\nInherited from [BaseAugmentation](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation): [augment_masks()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_masks), [augment_sample()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_sample), [augment_samples()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_samples), [augment_targets()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.augment_targets), [batch_augment()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.batch_augment), [call()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.call), [get_config()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.get_config), [get_random_transformations()](/helia-edge/reference/api/helia_edge/layers/preprocessing/base_augmentation/#helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation.get_random_transformations).",
          "examples": [],
          "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation2D",
          "kind": "class",
          "language": "python",
          "members": [
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.layers.preprocessing.base_augmentation.BaseAugmentation2D.NDIMS",
              "kind": "constant",
              "language": "python",
              "members": [],
              "name": "NDIMS",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "NDIMS = 4",
              "source": {
                "line": 250,
                "path": "helia_edge/layers/preprocessing/base_augmentation.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L250"
              },
              "summary": ""
            }
          ],
          "name": "BaseAugmentation2D",
          "params": [],
          "raises": [],
          "returns": [],
          "signature": "BaseAugmentation2D(\n    seed=None,\n    auto_vectorize=False,\n    data_format=None,\n    device='cpu',\n    aligned_targets=None,\n    aligned_masks=None,\n    **kwargs={},\n)",
          "source": {
            "line": 247,
            "path": "helia_edge/layers/preprocessing/base_augmentation.py",
            "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/preprocessing/base_augmentation.py#L247"
          },
          "summary": "Two-dimensional images with optional batch axis."
        }
      ]
    }
  ],
  "name": "helia_edge"
}