skainet-lang-core/sk.ainet.lang.tensor.dsl/MLPWeights/xavier

xavier

fun xavier(inputSize: Int, outputSize: Int, random: Random = Random.Default): TensorInitializer<T, V>(source)

Creates a weight matrix for a linear layer with Xavier/Glorot initialization

Parameters

inputSize

number of input features

outputSize

number of output features