package torch

  1. Overview
  2. Docs
package torch
Legend:
Library
Module
Module type
Parameter
Class
Class type

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_masked_fill : (t Ctypes_static.ptr -> t -> t -> scalar -> unit F.return) F.result
val stubs_masked_fill_ : (t Ctypes_static.ptr -> t -> t -> scalar -> unit F.return) F.result
val stubs_masked_fill_scalar_out : (t Ctypes_static.ptr -> t -> t -> t -> scalar -> unit F.return) F.result
val stubs_masked_fill_tensor : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_masked_fill_tensor_ : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_masked_fill_tensor_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> unit F.return) F.result
val stubs_masked_scatter : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_masked_scatter_ : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_masked_scatter_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> unit F.return) F.result
val stubs_masked_select : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs_masked_select_backward : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_masked_select_out : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_matmul : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs_matmul_out : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_matrix_exp : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs_matrix_exp_backward : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs_matrix_h : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs_matrix_power : (t Ctypes_static.ptr -> t -> int64 -> unit F.return) F.result
val stubs_matrix_power_out : (t Ctypes_static.ptr -> t -> t -> int64 -> unit F.return) F.result
val stubs_max : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs_max_dim : (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result
val stubs_max_dim_max : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> int -> unit F.return) F.result
val stubs_max_other : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs_max_out : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_max_pool1d : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_max_pool1d_with_indices : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_max_pool2d : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_max_pool2d_with_indices : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_max_pool2d_with_indices_backward : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> t -> unit F.return) F.result
val stubs_max_pool2d_with_indices_backward_grad_input : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> t -> unit F.return) F.result
val stubs_max_pool2d_with_indices_out : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_max_pool3d : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_max_pool3d_with_indices : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_max_pool3d_with_indices_backward : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> t -> unit F.return) F.result
val stubs_max_pool3d_with_indices_backward_grad_input : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> t -> unit F.return) F.result
val stubs_max_pool3d_with_indices_out : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_max_unpool2d : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs_max_unpool2d_out : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs_max_unpool3d : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs_max_unpool3d_out : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs_maximum : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs_maximum_out : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_mean : (t Ctypes_static.ptr -> t -> int -> unit F.return) F.result
val stubs_mean_dim : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs_mean_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int -> int -> unit F.return) F.result
val stubs_median : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs_median_dim : (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result
val stubs_median_dim_values : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> int -> unit F.return) F.result
val stubs_median_out : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs_meshgrid : (t Ctypes_static.ptr -> int -> t Ctypes_static.ptr F.return) F.result
val stubs_meshgrid_indexing : (t Ctypes_static.ptr -> int -> string -> t Ctypes_static.ptr F.return) F.result
val stubs_mh : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs_min : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs_min_dim : (t Ctypes_static.ptr -> t -> int64 -> int -> unit F.return) F.result
val stubs_min_dim_min : (t Ctypes_static.ptr -> t -> t -> t -> int64 -> int -> unit F.return) F.result
val stubs_min_other : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs_min_out : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_minimum : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs_minimum_out : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_miopen_batch_norm : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> int -> float -> float -> unit F.return) F.result
val stubs_miopen_batch_norm_backward : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t -> float -> unit F.return) F.result
val stubs_miopen_batch_norm_backward_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t -> t -> t -> t -> float -> unit F.return) F.result
val stubs_miopen_batch_norm_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t -> t -> int -> float -> float -> unit F.return) F.result
val stubs_miopen_convolution : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> int -> int -> unit F.return) F.result
val stubs_miopen_convolution_add_relu : (t Ctypes_static.ptr -> t -> t -> t -> scalar -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
val stubs_miopen_convolution_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> int -> int -> unit F.return) F.result
val stubs_miopen_convolution_relu : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
val stubs_miopen_convolution_transpose : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> int -> int -> unit F.return) F.result
val stubs_miopen_convolution_transpose_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> int -> int -> unit F.return) F.result
val stubs_miopen_depthwise_convolution : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> int -> int -> unit F.return) F.result
val stubs_miopen_depthwise_convolution_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> int -> int -> unit F.return) F.result
val stubs_miopen_rnn : (t Ctypes_static.ptr -> t -> t Ctypes_static.ptr -> int -> int64 -> t -> t -> int64 -> int64 -> int64 -> int -> float -> int -> int -> int64 Ctypes_static.ptr -> int -> t -> unit F.return) F.result
val stubs_miopen_rnn_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> t -> t Ctypes_static.ptr -> int -> int64 -> t -> t -> int64 -> int64 -> int64 -> int -> float -> int -> int -> int64 Ctypes_static.ptr -> int -> t -> unit F.return) F.result
val stubs_mish : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs_mish_ : (t Ctypes_static.ptr -> t -> unit F.return) F.result
val stubs_mish_backward : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs_mish_out : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs_mkldnn_adaptive_avg_pool2d : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs_mkldnn_adaptive_avg_pool2d_backward : (t Ctypes_static.ptr -> t -> t -> unit F.return) F.result
val stubs_mkldnn_adaptive_avg_pool2d_backward_out : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_mkldnn_adaptive_avg_pool2d_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> unit F.return) F.result
val stubs_mkldnn_convolution : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
val stubs_mkldnn_convolution_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
val stubs_mkldnn_linear : (t Ctypes_static.ptr -> t -> t -> t -> unit F.return) F.result
val stubs_mkldnn_linear_backward_input : (t Ctypes_static.ptr -> int64 Ctypes_static.ptr -> int -> t -> t -> unit F.return) F.result
val stubs_mkldnn_linear_backward_input_out : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> t -> t -> unit F.return) F.result
val stubs_mkldnn_linear_backward_weights : (t Ctypes_static.ptr -> t -> t -> t -> int -> unit F.return) F.result
val stubs_mkldnn_linear_backward_weights_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> t -> int -> unit F.return) F.result
val stubs_mkldnn_linear_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> unit F.return) F.result
val stubs_mkldnn_max_pool2d : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_mkldnn_max_pool2d_backward : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_mkldnn_max_pool2d_backward_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_mkldnn_max_pool2d_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_mkldnn_max_pool3d : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_mkldnn_max_pool3d_backward : (t Ctypes_static.ptr -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_mkldnn_max_pool3d_backward_out : (t Ctypes_static.ptr -> t -> t -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_mkldnn_max_pool3d_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int -> unit F.return) F.result
val stubs_mkldnn_reorder_conv2d_weight : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
val stubs_mkldnn_reorder_conv2d_weight_out : (t Ctypes_static.ptr -> t -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
val stubs_mkldnn_reorder_conv3d_weight : (t Ctypes_static.ptr -> t -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 Ctypes_static.ptr -> int -> int64 -> unit F.return) F.result
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct