torch_tensor Derived Type

type, public :: torch_tensor

Type for holding a Torch tensor.


Components

Type Visibility Attributes Name Initial
type(c_ptr), public :: p = c_null_ptr

pointer to the tensor in memory

Type-Bound Procedures

procedure, public :: get_device_index => torch_tensor_get_device_index

  • public function torch_tensor_get_device_index(tensor) result(device_index)

    Determines the device index of a tensor.

    Arguments

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

    Input tensor

    Return Value integer(kind=c_int)

    Device index of tensor

procedure, public :: get_device_type => torch_tensor_get_device_type

  • public function torch_tensor_get_device_type(tensor) result(device_type)

    Returns the device type of a tensor.

    Arguments

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

    Input tensor

    Return Value integer(kind=c_int)

    Device type of tensor

procedure, public :: get_dtype => torch_tensor_get_dtype

  • public function torch_tensor_get_dtype(tensor) result(dtype)

    Returns the data type of a tensor.

    Arguments

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

    Input tensor

    Return Value integer(kind=c_int)

    Data type of tensor

procedure, public :: get_rank => torch_tensor_get_rank

  • public function torch_tensor_get_rank(self) result(rank)

    Determines the rank of a tensor.

    Arguments

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

    Return Value integer(kind=int32)

    rank of tensor

procedure, public :: get_shape => torch_tensor_get_shape

  • public function torch_tensor_get_shape(self) result(sizes)

    Determines the shape of a tensor.

    Arguments

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

    Return Value integer(kind=c_long_long), pointer, (:)

    Pointer to tensor data

Source Code

  type torch_tensor
    type(c_ptr) :: p = c_null_ptr  !! pointer to the tensor in memory
  contains
    procedure :: get_rank => torch_tensor_get_rank
    procedure :: get_shape => torch_tensor_get_shape
    procedure :: get_dtype => torch_tensor_get_dtype
    procedure :: get_device_type => torch_tensor_get_device_type
    procedure :: get_device_index => torch_tensor_get_device_index
  end type torch_tensor