Function coaster_nn::frameworks::native::helper::relu_grad
source · pub fn relu_grad<T: Float>(x: T, dx: T) -> T
Expand description
Computes the ReLU Gradient on the CPU
pub fn relu_grad<T: Float>(x: T, dx: T) -> T
Computes the ReLU Gradient on the CPU