pub fn relu<T>(args: T) -> Twhere T: PartialOrd + Zero,
the relu activation function: $f(x) = \max(0, x)$