init method

  1. @override
void init([
  1. dynamic dims
])
override

Set initial values to weights and biases

dims parametr is input vector length

For weights w is used Matrix(m, n, seed) constructor with He-normal initialization

For biases b is used Matrix.zeros(n: n, m: 1) constructor

Implementation

@override
void init([dynamic dims]) {
  _prevDim = (dims as int);
  w = Matrix(n: units, m: _prevDim!, seed: 3);
  b = Matrix.zero(n: units, m: 1);
  wasInitialized = true;
}