search

brendanhasz / probflow

A Python package for building Bayesian models with TensorFlow or PyTorch
http://probflow.readthedocs.io
MIT License
170 stars 17 forks source link

Convolutional modules #25

Open brendanhasz opened 4 years ago

brendanhasz commented 4 years ago

Add Bayesian Conv1D and Conv2D modules.

Looks like maybe you can just get a sample from the filter's variational posteriors, then just pass the data and the sampled tensor through tf.ops.nn_ops.convolution.