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