Function that checks the provided activation function
Function that checks the provided activation function
vaeac_check_activation_func(activation_function)
Arguments
activation_function: An torch::nn_module() representing an activation function such as, e.g., torch::nn_relu() (default), torch::nn_leaky_relu(), torch::nn_selu(), or torch::nn_sigmoid().