mx.model.init.params
¶
Description¶
Parameter initialization
Usage¶
mx.model.init.params(symbol, input.shape, output.shape, initializer, ctx)
Arguments¶
Argument |
Description |
---|---|
|
The symbolic configuration of the neural network. |
|
The shape of the input for the neural network. |
|
The shape of the output for the neural network. It can be NULL. |
|
initializer object. The initialization scheme for parameters. |
|
mx.context. The devices used to perform initialization. |