Module NTDSL.O

val (*) : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val (*.) : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val (+) : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val threefry4x32 : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val uint4x32_to_prec_uniform : ?label:??? -> Tensor.t -> Tensor.t
val uint4x32_to_prec_uniform1 : ?label:??? -> Tensor.t -> Tensor.t
val (**.) : ?label:??? -> Tensor.t -> Base.float -> Tensor.t
val relu : ?label:??? -> Tensor.t -> Tensor.t
val stretch : ?label:??? -> Tensor.t -> Tensor.t
val sat01 : ?label:??? -> Tensor.t -> Tensor.t
val fma : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t -> Tensor.t
val (!.) : Base.float -> Tensor.t
val (!..) : ?label:??? -> int -> Tensor.t
val (!%) : ?label:??? -> Base.int64 -> Tensor.t
val (!@) : ?label:??? -> Ir.Indexing.static_symbol -> Tensor.t
val dim : ?label:??? -> Shape.delayed_var_ref -> Tensor.t
val (-) : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val (~-) : ?label:??? -> Tensor.t -> Tensor.t
val (/.) : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val (@|) : ?label:??? -> Tensor.t -> Idx.static_symbol -> Tensor.t
val exp : ?label:??? -> Tensor.t -> Tensor.t
val log : ?label:??? -> Tensor.t -> Tensor.t
val log2 : ?label:??? -> Tensor.t -> Tensor.t
val sin : ?label:??? -> Tensor.t -> Tensor.t
val cos : ?label:??? -> Tensor.t -> Tensor.t
val neg : ?label:??? -> Tensor.t -> Tensor.t
val sqrt : ?label:??? -> Tensor.t -> Tensor.t
val stop_gradient : ?label:??? -> Tensor.t -> Tensor.t
val recip : ?label:??? -> Tensor.t -> Tensor.t
val recip_sqrt : ?label:??? -> Tensor.t -> Tensor.t
val tanh : ?label:??? -> Tensor.t -> Tensor.t
val where : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t -> Tensor.t
val not : ?label:??? -> Tensor.t -> Tensor.t
val (<) : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val (<=) : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val (>) : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val (>=) : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val (=) : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val (<>) : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val embed_self_id : ?label:??? -> unit -> Tensor.t
val einsum : ?label:??? -> ?capture_dims:??? -> Base.string -> Tensor.t -> Tensor.t -> Tensor.t
val outer_sum : ?label:??? -> ?capture_dims:??? -> Base.string -> Tensor.t -> Tensor.t -> Tensor.t
val einsum1 : ?label:??? -> ?capture_dims:??? -> Base.string -> Tensor.t -> Tensor.t
val einmax1 : ?label:??? -> ?capture_dims:??? -> ?nonoverlapping:??? -> Base.string -> Tensor.t -> Tensor.t
val tropical : ?label:??? -> ?capture_dims:??? -> ?nonoverlapping:??? -> Base.string -> Tensor.t -> Tensor.t -> Tensor.t
val concat_sum : ?label:??? -> ?capture_dims:??? -> ?negated:??? -> Base.string -> Tensor.t Base.Array.t -> Tensor.t
val concat : ?label:??? -> ?capture_dims:??? -> ?negated:??? -> Base.string -> Tensor.t Base.Array.t -> Tensor.t
val stack : ?label:??? -> stack_axis -> Tensor.t Base.Array.t -> Tensor.t
val offsets : ?label:??? -> unit -> Tensor.t
val uniform : ?label:??? -> unit -> Tensor.t
val uniform_at : ?label:??? -> Tensor.t -> Tensor.t
val uniform1 : ?label:??? -> unit -> Tensor.t
val uniform_at1 : ?label:??? -> Tensor.t -> Tensor.t
val interleave : ?label:??? -> Tensor.t -> Tensor.t -> Tensor.t
val deinterleave_even : ?label:??? -> Tensor.t -> Tensor.t
val deinterleave_odd : ?label:??? -> Tensor.t -> Tensor.t