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