torch 0.14 · OCaml Package

Parameters

module F : Cstubs.FOREIGN

Signature

type t = unit Ctypes.ptr

val t : t Ctypes.typ

type scalar = unit Ctypes.ptr

val scalar : scalar Ctypes.typ

val stubs_trace : (t Ctypes_static.ptr -> t -> unit F.return) F.result

val stubs_trace_backward : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

val stubs_transpose : 
  (t Ctypes_static.ptr -> t -> int64 -> int64 -> unit F.return) F.result

val stubs_transpose_ : 
  (t Ctypes_static.ptr -> t -> int64 -> int64 -> unit F.return) F.result

val stubs_trapezoid : 
  (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result

val stubs_trapezoid_x : 
  (t Ctypes_static.ptr -> t -> t -> int64 -> unit F.return) F.result

val stubs_trapz : 
  (t Ctypes_static.ptr -> t -> t -> int64 -> unit F.return) F.result

val stubs_trapz_dx : 
  (t Ctypes_static.ptr -> t -> float -> int64 -> unit F.return) F.result

val stubs_triangular_solve : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs_triangular_solve_x : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    t ->
    int ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs_tril : (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result

val stubs_tril_ : (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result

val stubs_tril_indices : 
  (t Ctypes_static.ptr ->
    int64 ->
    int64 ->
    int64 ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs_tril_out : 
  (t Ctypes_static.ptr -> t -> t -> int64 -> unit F.return) F.result

val stubs_triplet_margin_loss : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    t ->
    float ->
    float ->
    float ->
    int ->
    int64 ->
    unit F.return)
    F.result

val stubs_triu : (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result

val stubs_triu_ : (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result

val stubs_triu_indices : 
  (t Ctypes_static.ptr ->
    int64 ->
    int64 ->
    int64 ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs_triu_out : 
  (t Ctypes_static.ptr -> t -> t -> int64 -> unit F.return) F.result

val stubs_true_divide : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs_true_divide_ : 
  (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs_true_divide_out : 
  (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result

val stubs_true_divide_scalar : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs_true_divide_scalar_ : 
  (t Ctypes_static.ptr -> t -> scalar -> unit F.return) F.result

val stubs_trunc : (t Ctypes_static.ptr -> t -> unit F.return) F.result

val stubs_trunc_ : (t Ctypes_static.ptr -> t -> unit F.return) F.result

val stubs_trunc_out : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs_type_as : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs_unbind : (t -> int64 -> t Ctypes_static.ptr F.return) F.result

val stubs_unflatten : 
  (t Ctypes_static.ptr ->
    t ->
    int64 ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

val stubs_unflatten_dense_tensors : 
  (t -> t Ctypes_static.ptr -> int -> t Ctypes_static.ptr F.return) F.result

val stubs_unfold : 
  (t Ctypes_static.ptr ->
    t ->
    int64 ->
    int64 ->
    int64 ->
    unit F.return)
    F.result

val stubs_unfold_backward : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int64 ->
    int64 ->
    unit F.return)
    F.result

val stubs_uniform_ : 
  (t Ctypes_static.ptr -> t -> float -> float -> unit F.return) F.result

val stubs_unique_consecutive : 
  (t Ctypes_static.ptr -> t -> int -> int -> int64 -> unit F.return) F.result

val stubs_unique_dim : 
  (t Ctypes_static.ptr ->
    t ->
    int64 ->
    int ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs_unique_dim_consecutive : 
  (t Ctypes_static.ptr -> t -> int64 -> int -> int -> unit F.return) F.result

val stubs_unsafe_chunk : 
  (t -> int64 -> int64 -> t Ctypes_static.ptr F.return) F.result

val stubs_unsafe_split : 
  (t -> int64 -> int64 -> t Ctypes_static.ptr F.return) F.result

val stubs_unsafe_split_with_sizes : 
  (t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    t Ctypes_static.ptr F.return)
    F.result

val stubs_unsqueeze : 
  (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result

val stubs_unsqueeze_ : 
  (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result

val stubs_upsample_bicubic2d : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_bicubic2d_backward : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_bicubic2d_backward_grad_input : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_bicubic2d_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_bilinear2d : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_bilinear2d_backward : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_bilinear2d_backward_grad_input : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_bilinear2d_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_linear1d : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_linear1d_backward : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_linear1d_backward_grad_input : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_linear1d_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest1d : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest1d_backward : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest1d_backward_grad_input : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest1d_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest2d : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest2d_backward : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest2d_backward_grad_input : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest2d_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest3d : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest3d_backward : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest3d_backward_grad_input : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_nearest3d_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    float ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_trilinear3d : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_trilinear3d_backward : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_trilinear3d_backward_grad_input : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_upsample_trilinear3d_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    float ->
    float ->
    float ->
    unit F.return)
    F.result

val stubs_value_selecting_reduction_backward : 
  (t Ctypes_static.ptr ->
    t ->
    int64 ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs_values : (t Ctypes_static.ptr -> t -> unit F.return) F.result

val stubs_vander : 
  (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result

val stubs_var : (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs_var_correction : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    unit F.return)
    F.result

val stubs_var_correction_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    unit F.return)
    F.result

val stubs_var_dim : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs_var_mean : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs_var_mean_correction : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int64 ->
    int ->
    unit F.return)
    F.result

val stubs_var_mean_dim : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs_var_out : 
  (t Ctypes_static.ptr ->
    t ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    int ->
    int ->
    unit F.return)
    F.result

val stubs_vdot : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs_vdot_out : 
  (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result

val stubs_view : 
  (t Ctypes_static.ptr ->
    t ->
    int64 Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

val stubs_view_as : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs_view_as_complex : 
  (t Ctypes_static.ptr -> t -> unit F.return) F.result

val stubs_view_as_real : (t Ctypes_static.ptr -> t -> unit F.return) F.result

val stubs_view_dtype : 
  (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result

val stubs_vsplit : (t -> int64 -> t Ctypes_static.ptr F.return) F.result

val stubs_vsplit_array : 
  (t ->
    int64 Ctypes_static.ptr ->
    int ->
    t Ctypes_static.ptr F.return)
    F.result

val stubs_vstack : 
  (t Ctypes_static.ptr -> t Ctypes_static.ptr -> int -> unit F.return) F.result

val stubs_vstack_out : 
  (t Ctypes_static.ptr ->
    t ->
    t Ctypes_static.ptr ->
    int ->
    unit F.return)
    F.result

val stubs_where : (t -> t Ctypes_static.ptr F.return) F.result

val stubs_where_scalar : 
  (t Ctypes_static.ptr -> t -> scalar -> scalar -> unit F.return) F.result

val stubs_where_scalarother : 
  (t Ctypes_static.ptr -> t -> t -> scalar -> unit F.return) F.result

val stubs_where_scalarself : 
  (t Ctypes_static.ptr -> t -> scalar -> t -> unit F.return) F.result

val stubs_where_self : 
  (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result

val stubs_xlogy : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs_xlogy_ : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result

val stubs_xlogy_outscalar_other : 
  (t Ctypes_static.ptr -> t -> t -> scalar -> unit F.return) F.result