multiply_real2d Module Function

module function multiply_real2d(a, b) result(c)

Multiply autodiff array by real array (element-wise)

Arguments

Type IntentOptional Attributes Name
class(array_type), intent(in), target :: a
real(kind=real32), intent(in), dimension(:,:) :: b

Return Value type(array_type), pointer


Source Code

  module function multiply_real2d(a, b) result(c)
    !! Multiply autodiff array by real array (element-wise)
    implicit none
    class(array_type), intent(in), target :: a
    real(real32), dimension(:,:), intent(in) :: b
    type(array_type), pointer :: c
    type(array_type), pointer :: b_array

    integer :: s, i

    c => a%create_result()
    do concurrent(s=1:size(a%val,2), i=1:size(a%val,1))
       c%val(i,s) = a%val(i,s) * b(i,s)
    end do

    c%get_partial_left => get_partial_multiply_left
    if(a%requires_grad)then
       c%requires_grad = .true.
       c%is_forward = a%is_forward
       c%operation = 'multiply_real2d'
       c%left_operand => a
       c%owns_left_operand = a%is_temporary
    end if
    allocate(b_array)
    b_array%is_scalar = .false.
    b_array%requires_grad = .false.
    call b_array%allocate(array_shape=shape(b))
    b_array%val(:,:) = b
    c%right_operand => b_array
    c%owns_right_operand = .true.
  end function multiply_real2d