Method ConditionedFlowMLPModule

src/framework/modules/conditioning_modules.cpp:325–331

}

ConditionedFlowMLPModule::ConditionedFlowMLPModule(ConditionedFlowMLPConfig config) : config_(config) {
    require_positive(config_.input_size, "ConditionedFlowMLPConfig.input_size");
    require_positive(config_.hidden_size, "ConditionedFlowMLPConfig.hidden_size");
    require_positive(config_.condition_size, "ConditionedFlowMLPConfig.condition_size");
    require_positive(config_.output_size, "ConditionedFlowMLPConfig.output_size");
    require_positive(config_.layers, "ConditionedFlowMLPConfig.layers");
}

core::TensorValue ConditionedFlowMLPModule::build(
    core::ModuleBuildContext & ctx,