View source on GitHub |
Returns default initialization bounds depending on layer output bounds.
tfl.kronecker_factored_lattice_lib.default_init_params(
output_min, output_max
)
Args | |
---|---|
output_min
|
None or minimum layer output. |
output_max
|
None or maximum layer output. |