package tensorflow

You can search for identifiers within the package.

in-package search v0.2.0

tensorflow
- Tensorflow

Legend:
Library
Module
Module type
Parameter
Class
Class type

val with_shape : 
  f:(shape:'dim shape -> 'a) ->
  (('dim t -> 'a) -> 'b) ->
  'b Core_kernel.Std.Staged.t

val dense : shape:int -> (_1d t -> _1d t) Core_kernel.Std.Staged.t

val conv2d : 
  filter:(int * int) ->
  out_channels:int ->
  strides:(int * int) ->
  padding:[ `same | `valid ] ->
  (_3d t -> _3d t) Core_kernel.Std.Staged.t