layer_dense_module: Function to create custom nn_linear module to overwrite reset_parameters
Description
Function to create custom nn_linear module to overwrite reset_parameters
Usage
layer_dense_module(kernel_initializer)
Arguments
- kernel_initializer
string; initializer used to reset_parameters