-
贝叶斯火炬是一个库,其中包含了神经网络层和实用工具,它对PyTorch的核心进行了扩展,使得用户能够在贝叶斯深度神经网络中执行随机变异推断
资源介绍
贝叶斯-火炬:贝叶斯神经网络层用于不确定性估计
| | |
贝叶斯层和实用程序在PyTorch中执行随机变分推断
贝叶斯火炬是神经网络层和实用程序的库,它扩展了PyTorch的核心,使用户能够在贝叶斯深度神经网络中执行随机变异推断。 通过简单地将确定性层替换为贝叶斯层,贝叶斯-火炬被设计为灵活,无缝地将确定性深度神经网络体系结构扩展为相应的贝叶斯形式。
该存储库具有以下贝叶斯层的实现:
[ ]
LinearVariational
Conv1dVariational, Conv2dVariational, Conv3dVariational, ConvTranspose1dVariational, ConvTranspose2dVariational, ConvTranspose3dVariational
LSTMVariational
[ ]
LinearFlipo
- 上一篇: 时间序列MATLAB程序.docx
- 下一篇: pytorch实现softmax多重分类