package tensorflow

You can search for identifiers within the package.

in-package search v0.2.0

package tensorflow

tensorflow
- Tensorflow
  - Cell
  - Fnn
    
    Input_id
    
    Loss
    
    Model
    
    Optimizer
    
    Shape
  - Gradients
  - Layer
  - Mnist_helper
  - Node
    
    Attr_list
    
    Dim
    
    Tensor
    
    Type
    
    Weak_table
  - Ops
    
    Op_names
    
    Placeholder
  - Ops_generated
    
    Op_names
  - Ops_gradients
  - Ops_manual
    
    Placeholder
  - Optimizers
  - Registered_gradients
  - Session
    
    Input
    
    Output
  - Tensor
  - Var
  - Wrapper
    
    Graph
    
    Session
    
    Session_options
    
    Status

Legend:
Library
Module
Module type
Parameter
Class
Class type

type _1d

type _2d

type _3d

module Shape : sig ... end

module Input_id : sig ... end

type 'a t

type init = [

| `const of float
| `normal of float
| `truncated_normal of float

]

val shape : 'a t -> 'a Shape.t

val input : shape:'a Shape.t -> 'a t * Input_id.t

val const : float -> shape:'a Shape.t -> 'a t

val sigmoid : 'a t -> 'a t

val tanh : 'a t -> 'a t

val relu : 'a t -> 'a t

val softmax : 'a t -> 'a t

val (+) : 'a t -> 'a t -> 'a t

val (-) : 'a t -> 'a t -> 'a t

val (*) : 'a t -> 'a t -> 'a t

val dense : ?w_init:init -> ?b_init:init -> int -> _1d t -> _1d t

val conv2d : 
  ?w_init:init ->
  ?b_init:init ->
  filter:(int * int) ->
  out_channels:int ->
  strides:(int * int) ->
  padding:[ `same | `valid ] ->
  unit ->
  _3d t ->
  _3d t

val dense' : 
  ?w_init:init ->
  ?b_init:init ->
  int ->
  (_1d t -> _1d 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 t -> _3d t) Core_kernel.Std.Staged.t

val max_pool : 
  _3d t ->
  filter:(int * int) ->
  strides:(int * int) ->
  padding:[ `same | `valid ] ->
  _3d t

val reshape : 'b t -> shape:'a Shape.t -> 'a t

val flatten : 'a t -> _1d t

val split : _2d t -> _1d t list

val concat : _1d t list -> _2d t

module Optimizer : sig ... end

module Loss : sig ... end

module Model : sig ... end