MLPBias

class MLPBias<T : DType, V>(builder: TensorBuilder<T, V>)(source)

Creates bias tensors for MLP layers

Constructors

Link copied to clipboard
constructor(builder: TensorBuilder<T, V>)

Functions

Link copied to clipboard
fun constant(size: Int, value: Number): TensorInitializer<T, V>

Creates a bias vector initialized with a constant value

Link copied to clipboard
fun normal(size: Int, mean: Float = 0.0f, std: Float = 0.01f, random: Random = Random.Default): TensorInitializer<T, V>

Creates a bias vector initialized with normal distribution

Link copied to clipboard
fun ones(size: Int): TensorInitializer<T, V>

Creates a bias vector initialized with ones

Link copied to clipboard
fun zeros(size: Int): TensorInitializer<T, V>

Creates a bias vector initialized with zeros (most common)