% Generated by roxygen2: do not edit by hand
% Please edit documentation in R/nn-kaf.R
\name{nn_kaf}
\alias{nn_kaf}
\title{Kolmogorov-Arnold Fourier Network}
\usage{
nn_kaf(
  layers,
  num_grids = 8,
  dropout = 0,
  use_layernorm = TRUE,
  activation_expectation = 1.64,
  fourier_init_scale = 0.01
)
}
\arguments{
\item{layers}{Integer vector. Network architecture, including input and
output dimensions. For example, \code{c(10, 64, 64, 1)}.}

\item{num_grids}{Integer. Number of Fourier frequencies per KAF layer.}

\item{dropout}{Numeric. Dropout probability.}

\item{use_layernorm}{Logical. Whether to apply layer normalization before
the Fourier feature block.}

\item{activation_expectation}{Numeric. Scaling constant for Fourier
initialization.}

\item{fourier_init_scale}{Numeric. Initial scale of the Fourier component.}
}
\value{
A torch nn_module.
}
\description{
Torch module implementing a stacked Kolmogorov-Arnold Fourier Network.
}
