File: arrays_op_19.f90

package info (click to toggle)
lfortran 0.45.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 46,332 kB
  • sloc: cpp: 137,068; f90: 51,260; python: 6,444; ansic: 4,277; yacc: 2,285; fortran: 806; sh: 524; makefile: 30; javascript: 15
file content (24 lines) | stat: -rw-r--r-- 434 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
 
program array_op_19
implicit none

real :: x(10, 10), sx(10, 10)
x = 2.0
sx = softmax(x)
print *, sx
if( any(sx - 0.1 > 1e-6) ) error stop

contains

function softmax(x) result(y)
    real, intent(in) :: x(:, :)
    real :: y(size(x, 1), size(x, 2))
    integer :: i
    real :: s

    do i = 1, size(x, 2)
        y(:, i) = exp(x(:, i) - maxval(x(:, i)))
        y(:, i) = y(:, i) / sum(y(:, i))
    end do
end function

end program