multiply_logical Module Function

  module function multiply_logical(a, b) result(c)
    !! Multiply two logical arrays (element-wise)
    implicit none
    class(array_type), intent(in), target :: a
    logical, dimension(:,:), intent(in) :: b
    type(array_type), pointer :: c

    integer :: s, i

    c => a%create_result()
    do concurrent(s=1:size(a%val,2), i=1:size(a%val,1))
       if(b(i,s))then
          c%val(i,s) = a%val(i,s)
       else
          c%val(i,s) = 0._real32
       end if
    end do
    c%mask = b

    c%get_partial_left => get_partial_multiply_logical_left
    c%get_partial_left_val => get_partial_multiply_logical_left_val
    if(a%requires_grad)then
       c%requires_grad = .true.
       c%is_forward = a%is_forward
       c%operation = 'multiply_logical'
       c%left_operand => a
       c%owns_left_operand = a%is_temporary
    end if

  end function multiply_logical