createAkn

fun <T : DType, V> createAkn(executionContext: ExecutionContext, dtype: KClass<T>, inputDim: Int, outputDim: Int, gridSize: Int = 16, degree: Int = 3, useBias: Boolean = true, useResidual: Boolean = false, name: String = "akn", baseActivation: (Tensor<T, V>) -> Tensor<T, V> = { it }, weightsInit: WeightsScope<T, V>.(Shape) -> Tensor<T, V> = { randn(std = 0.02f) }, basisInit: WeightsScope<T, V>.(Shape) -> Tensor<T, V> = { uniform(min = -0.5f, max = 0.5f) }, biasInit: BiasScope<T, V>.(Shape) -> Tensor<T, V> = { zeros() }): Akn<T, V>(source)

Factory to create an AKN module directly (without the kanLayer DSL helper). Mirrors defaults and initializer hooks used by the DSL.