package tensorflow

  1. Overview
  2. Docs
val with_shape : f:(shape:'dim Shape.t -> type_:'c Node.Type.t -> 'a) -> ((('dim, 'c) t -> 'a) -> 'b) -> 'b Core_kernel.Std.Staged.t
val dense : ?w_init:init -> ?b_init:init -> shape:int -> unit -> ((_1d, [< `double | `float ] as 'b) t -> (_1d, 'b) t) Core_kernel.Std.Staged.t
val conv2d : ?w_init:init -> ?b_init:init -> filter:(int * int) -> out_channels:int -> strides:(int * int) -> padding:[ `same | `valid ] -> unit -> ((_3d, [< `double | `float ] as 'b) t -> (_3d, 'b) t) Core_kernel.Std.Staged.t
OCaml

Innovation. Community. Security.