torch_tensor_power_real32 Function

public function torch_tensor_power_real32(tensor, power) result(output)

Uses

Overloads exponentiation operator for a tensor and a scalar of type real32

Arguments

Type IntentOptional Attributes Name
type(torch_tensor), intent(in) :: tensor
real(kind=real32), intent(in), target :: power

Return Value type(torch_tensor)


Source Code

  function torch_tensor_power_real32(tensor, power) result(output)
    use, intrinsic :: iso_c_binding, only : c_loc
    use, intrinsic :: iso_fortran_env, only : real32
    type(torch_tensor), intent(in) :: tensor
    real(kind=real32), target, intent(in) :: power
    type(torch_tensor) :: output

    interface
      function torch_tensor_power_float_c(tensor_c, power_c) result(output_c)        &
          bind(c, name = 'torch_tensor_power_float')
        use, intrinsic :: iso_c_binding, only : c_ptr, c_float
        implicit none
        type(c_ptr), value, intent(in) :: tensor_c
        type(c_ptr), value, intent(in) :: power_c
        type(c_ptr) :: output_c
      end function torch_tensor_power_float_c
    end interface

    output%p = torch_tensor_power_float_c(tensor%p, c_loc(power))
  end function torch_tensor_power_real32