relu_prime Function

  1. mod_activation.f90
  2. mod_activation
  3. relu_prime

public pure function relu_prime(x) result(res)

First derivative of the REctified Linear Unit (RELU) activation function.

Arguments

Type IntentOptional AttributesName
real(kind=rk), intent(in) :: x(:)

Return Value real(kind=rk) (size(x))


Contents

Source Code

relu_prime

Source Code

  pure function relu_prime(x) result(res)
    !! First derivative of the REctified Linear Unit (RELU) activation function.
    real(rk), intent(in) :: x(:)
    real(rk) :: res(size(x))
    where (x > 0)
      res = 1
    elsewhere
      res = 0
    end where
  end function relu_prime