get_partial_elem_scale_scale_val Subroutine

  pure subroutine get_partial_elem_scale_scale_val(this, upstream_grad, output)
    !! d(out)/d(scale): upstream * input (element-wise, per sample)
    implicit none

    ! Arguments
    class(array_type), intent(in) :: this
    !! Forward result node containing saved operands
    real(real32), dimension(:,:), intent(in)  :: upstream_grad
    !! Upstream gradient values
    real(real32), dimension(:,:), intent(out) :: output
    !! Output gradient values for scale tensor

    ! Local variables
    integer :: i, s
    !! Feature and sample indices

    do concurrent(s = 1:size(output,2), i = 1:size(output,1))
       output(i, s) = upstream_grad(i, s) * this%left_operand%val(i, s)
    end do

  end subroutine get_partial_elem_scale_scale_val