Continuing the discussion from How are layer weights and biases initialized by default?:
You might wanna check out this section of the documentation