softmax Interface

public interface softmax

softmax function. Available for ranks 1 to 4


Functions

private pure module function softmax_r1_dp(x) result(y)

Arguments

Type IntentOptional Attributes Name
real(kind=dp), intent(in) :: x(:)

Return Value real(kind=dp), (size(x))

private pure module function softmax_r1_sp(x) result(y)

(Specification)

Arguments

Type IntentOptional Attributes Name
real(kind=sp), intent(in) :: x(:)

Return Value real(kind=sp), (size(x))

private pure module function softmax_r2_dp(x, dim) result(y)

Arguments

Type IntentOptional Attributes Name
real(kind=dp), intent(in) :: x(:,:)
integer, intent(in), optional :: dim

Return Value real(kind=dp), (size(x,1),size(x,2))

private pure module function softmax_r2_sp(x, dim) result(y)

Arguments

Type IntentOptional Attributes Name
real(kind=sp), intent(in) :: x(:,:)
integer, intent(in), optional :: dim

Return Value real(kind=sp), (size(x,1),size(x,2))

private pure module function softmax_r3_dp(x, dim) result(y)

Arguments

Type IntentOptional Attributes Name
real(kind=dp), intent(in) :: x(:,:,:)
integer, intent(in), optional :: dim

Return Value real(kind=dp), (size(x,1),size(x,2),size(x,3))

private pure module function softmax_r3_sp(x, dim) result(y)

Arguments

Type IntentOptional Attributes Name
real(kind=sp), intent(in) :: x(:,:,:)
integer, intent(in), optional :: dim

Return Value real(kind=sp), (size(x,1),size(x,2),size(x,3))