mxfusion.components.functions.mxfusion_gluon_function¶
Members¶
-
class
mxfusion.components.functions.mxfusion_gluon_function.
MXFusionGluonFunction
(block, nOutputs, dtype=None, broadcastable=False)¶ Bases:
object
The wrapper of a MXNet Gluon block in MXFusion. It automatically fetches all the Gluon parameters in its ParameterDict. When this function wrapper is called in Model definition, it returns a factor corresponding to the function evaluation.
Parameters: - block (mxnet.gluon.Blockk or mxnet.gluon.HybridBlock) – The MXNet Gluon block to be wrapped.
- nOutputs (int) – The number of output variables of the Gluon block.
- dtype (numpy.float32 or numpy.float64) – the data type of float point numbers used in the Gluon block.
- broadcastable (boolean) – Whether the function supports broadcasting with the additional dimension for samples.
-
collect_internal_parameters
()¶ Return the parameters of the MXNet Gluon block that have not been set a prior distribution.
Returns: the parameters of the MXNet Gluon block without a prior distribution. Return type: MXNet.gluon.ParameterDict
-
collect_params
()¶ Return a variable set / dict. Used for the function.collect_params.set_prior() functionality.
-
name
¶