Exception: ERROR: Unsupported layer type: BayesianDropout

[shiyishiguang]

When I reproduce the project: https://github.com/os-hxfan/BayesNN_FPGA When running, run it in the Hardware_ Artifact/ayes_hw directory python3 hls4ml_pred.py --load_model ./exp_mnist_bayes_lenet/mnist_bayes_lenet_spt_3samples_mc --dataset mnist --gpus 0,1,2,3 --spt_tmp_opt --mc_samples 3 --dropout_type mc， The error is as follows: raise Exception('ERROR: Unsupported layer type: {}'.format(keras_layer['class_name']))
Exception: ERROR: Unsupported layer type: BayesianDropout。
May I ask if hls4ml supports custom layers?

[bo3z]

You can check out the Extension API: https://fastmachinelearning.org/hls4ml/advanced/extension.html