% Generated by roxygen2: do not edit by hand
% Please edit documentation in R/diagnostics.R
\name{plot_kaf_scales}
\alias{plot_kaf_scales}
\title{Plot KAF branch scales}
\usage{
plot_kaf_scales(object, layer = 1, type = c("ratio", "branch"), ...)
}
\arguments{
\item{object}{A fitted object returned by \code{kaf_fit()}, or a raw \code{nn_kaf}
torch module.}

\item{layer}{Integer. Layer to inspect.}

\item{type}{Character. Either \code{"ratio"} or \code{"branch"}.}

\item{...}{Additional arguments passed to base plotting functions.}
}
\value{
Invisibly returns the extracted scale data.
}
\description{
Visualizes the learned base/GELU and Fourier branch scales for a selected KAF
layer.
}
