he

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

Creates a weight matrix for a linear layer with He initialization (good for ReLU)

inputSize

number of input features

outputSize

number of output features