skainet-kan/sk.ainet.lang.kan/KAN

KAN

interface KAN<T : DType, V> : NetworkDslItem(source)

DSL surface for configuring a Kolmogorov–Arnold Network layer.

Inheritors

Properties

abstract var baseActivation: (Tensor<T, V>) -> Tensor<T, V>

abstract var degree: Int

abstract var gridSize: Int

abstract var outputDim: Int

abstract var useBias: Boolean

abstract var useResidual: Boolean

Functions

abstract fun basis(initBlock: WeightsScope<T, V>.(Shape) -> Tensor<T, V>)

abstract fun bias(initBlock: BiasScope<T, V>.(Shape) -> Tensor<T, V>)

abstract fun weights(initBlock: WeightsScope<T, V>.(Shape) -> Tensor<T, V>)