KanDslImpl

class KanDslImpl<T : DType, V>(val executionContext: ExecutionContext, dtype: KClass<T>, initialOutputDim: Int, id: String) : KAN<T, V> (source)

Constructors

constructor(executionContext: ExecutionContext, dtype: KClass<T>, initialOutputDim: Int, id: String)

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

open override var degree: Int

open override var gridSize: Int

open override var outputDim: Int

open override var useBias: Boolean

open override var useResidual: Boolean

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

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

fun create(inputDim: Int, mixingWeights: Tensor<T, V>, basis: Tensor<T, V>, bias: Tensor<T, V>?): KanLayer<T, V>

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