{
  "$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": "",
      "name": "squeeze_excite",
      "path": "helia_edge.layers.squeeze_excite",
      "submodules": [],
      "summary": "",
      "symbols": [
        {
          "description": "Squeeze & excite functional layer\n\nImplements Squeeze and Excite block as in\n[Squeeze-and-Excitation Networks](https://arxiv.org/pdf/1709.01507.pdf).",
          "examples": [],
          "id": "helia_edge.layers.squeeze_excite.se_layer",
          "kind": "function",
          "language": "python",
          "members": [],
          "name": "se_layer",
          "params": [
            {
              "default": "8",
              "description": "Expansion ratio. Defaults to 8.",
              "name": "ratio",
              "type": "Expansion ratio"
            },
            {
              "default": "None",
              "description": "Block name. Defaults to None.",
              "name": "name",
              "type": "str | None"
            },
            {
              "default": "'relu6'",
              "description": "Squeeze activation. Defaults to \"relu6\".",
              "name": "squeeze_activation",
              "type": "str | Callable"
            },
            {
              "default": "'hard_sigmoid'",
              "description": "Excite activation. Defaults to \"hard_sigmoid\".",
              "name": "excite_activation",
              "type": "str | Callable"
            }
          ],
          "raises": [],
          "returns": [
            {
              "description": "keras.Layer: Functional SE layer",
              "type": "keras.Layer"
            }
          ],
          "signature": "se_layer(\n    ratio: int = 8,\n    name: str | None = None,\n    squeeze_activation: str | Callable = 'relu6',\n    excite_activation: str | Callable = 'hard_sigmoid',\n) -> keras.Layer",
          "source": {
            "line": 6,
            "path": "helia_edge/layers/squeeze_excite.py",
            "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/layers/squeeze_excite.py#L6"
          },
          "summary": "Squeeze & excite functional layer"
        }
      ]
    }
  ],
  "name": "helia_edge"
}
