BDense: Create a Dense layer that has a standard normal prior for...

In enweg/quickbnnr: Quick way to implement Bayesian Neural Networks in R and estimating these using Julia's Turing

Create a Dense layer that has a standard normal prior for weights and biases

Description

Create a Dense layer that has a standard normal prior for weights and biases

Usage

BDense(
  in_size,
  out_size,
  activation = c("identity", "sigmoid", "tanh", "relu")
)

Arguments

in_size	input size
out_size	output size
activation	activation function

enweg/quickbnnr documentation built on April 15, 2022, 3:29 a.m.