pub fn relu_grad<T: Float>(x: T, dx: T) -> T
Expand description

Computes the ReLU Gradient on the CPU