torch_tensor_get_shape Function

public function torch_tensor_get_shape(self) result(sizes)

Uses

  • proc~~torch_tensor_get_shape~~UsesGraph proc~torch_tensor_get_shape torch_tensor%torch_tensor_get_shape iso_c_binding iso_c_binding proc~torch_tensor_get_shape->iso_c_binding

Determines the shape of a tensor.

Type Bound

torch_tensor

Arguments

Type IntentOptional Attributes Name
class(torch_tensor), intent(in) :: self

Tensor to get the shape of

Return Value integer(kind=int64), allocatable, (:)

Array holding the shape of the tensor


Calls

proc~~torch_tensor_get_shape~~CallsGraph proc~torch_tensor_get_shape torch_tensor%torch_tensor_get_shape proc~torch_tensor_get_rank torch_tensor%torch_tensor_get_rank proc~torch_tensor_get_shape->proc~torch_tensor_get_rank

Called by

proc~~torch_tensor_get_shape~~CalledByGraph proc~torch_tensor_get_shape torch_tensor%torch_tensor_get_shape proc~torch_tensor_add torch_tensor_add proc~torch_tensor_add->proc~torch_tensor_get_shape proc~torch_tensor_assign torch_tensor_assign proc~torch_tensor_assign->proc~torch_tensor_get_shape proc~torch_tensor_backward_without_external_gradient torch_tensor_backward_without_external_gradient proc~torch_tensor_backward_without_external_gradient->proc~torch_tensor_get_shape proc~torch_tensor_divide torch_tensor_divide proc~torch_tensor_divide->proc~torch_tensor_get_shape proc~torch_tensor_multiply torch_tensor_multiply proc~torch_tensor_multiply->proc~torch_tensor_get_shape proc~torch_tensor_negative torch_tensor_negative proc~torch_tensor_negative->proc~torch_tensor_get_shape proc~torch_tensor_power_int16 torch_tensor_power_int16 proc~torch_tensor_power_int16->proc~torch_tensor_get_shape proc~torch_tensor_power_int32 torch_tensor_power_int32 proc~torch_tensor_power_int32->proc~torch_tensor_get_shape proc~torch_tensor_power_int64 torch_tensor_power_int64 proc~torch_tensor_power_int64->proc~torch_tensor_get_shape proc~torch_tensor_power_int8 torch_tensor_power_int8 proc~torch_tensor_power_int8->proc~torch_tensor_get_shape proc~torch_tensor_power_real32 torch_tensor_power_real32 proc~torch_tensor_power_real32->proc~torch_tensor_get_shape proc~torch_tensor_power_real64 torch_tensor_power_real64 proc~torch_tensor_power_real64->proc~torch_tensor_get_shape proc~torch_tensor_subtract torch_tensor_subtract proc~torch_tensor_subtract->proc~torch_tensor_get_shape proc~torch_tensor_to torch_tensor_to proc~torch_tensor_to->proc~torch_tensor_get_shape interface~assignment (=) assignment (=) interface~assignment (=)->proc~torch_tensor_assign interface~operator (+) operator (+) interface~operator (+)->proc~torch_tensor_add interface~operator (-) operator (-) interface~operator (-)->proc~torch_tensor_negative interface~operator (-)->proc~torch_tensor_subtract interface~operator (ASTERISK) operator (*) interface~operator (ASTERISK)->proc~torch_tensor_multiply interface~operator (ASTERISKASTERISK) operator (**) interface~operator (ASTERISKASTERISK)->proc~torch_tensor_power_int16 interface~operator (ASTERISKASTERISK)->proc~torch_tensor_power_int32 interface~operator (ASTERISKASTERISK)->proc~torch_tensor_power_int64 interface~operator (ASTERISKASTERISK)->proc~torch_tensor_power_int8 interface~operator (ASTERISKASTERISK)->proc~torch_tensor_power_real32 interface~operator (ASTERISKASTERISK)->proc~torch_tensor_power_real64 interface~operator (SLASH) operator (/) interface~operator (SLASH)->proc~torch_tensor_divide interface~torch_tensor_backward torch_tensor_backward interface~torch_tensor_backward->proc~torch_tensor_backward_without_external_gradient

Source Code

  function torch_tensor_get_shape(self) result(sizes)
    use, intrinsic :: iso_c_binding, only : c_f_pointer, c_int64_t, c_ptr
    class(torch_tensor), intent(in) :: self         !! Tensor to get the shape of
    integer(kind=int64), allocatable :: sizes(:)    !! Array holding the shape of the tensor

    ! Local data
    integer(kind=int32) :: ndims(1)
    integer(kind=c_int64_t), pointer :: sizes_c_int64_ptr(:)
        !! Temporary pointer to Torch-owned memory containing c_int64
    type(c_ptr) :: cptr

    interface
      function torch_tensor_get_sizes_c(tensor_c) result(sizes_c) &
          bind(c, name = "torch_tensor_get_sizes")
        use, intrinsic :: iso_c_binding, only : c_ptr
        implicit none
        type(c_ptr), value, intent(in) :: tensor_c
        type(c_ptr) :: sizes_c
      end function torch_tensor_get_sizes_c
    end interface

    if (.not. c_associated(self%p)) then
      write(*,*) "Error :: tensor has not been constructed so its shape is unset"
      stop 1
    end if
    ndims(1) = self%get_rank()
    cptr = torch_tensor_get_sizes_c(self%p)
    call c_f_pointer(cptr, sizes_c_int64_ptr, ndims)

    ! Copy out of the Torch-owned memory so the result remains valid even if
    ! the tensor is subsequently deleted
    allocate(sizes(ndims(1)))
    sizes(:) = sizes_c_int64_ptr(:)
  end function torch_tensor_get_shape