package tensorflow

You can search for identifiers within the package.

in-package search v0.2.0

Legend:
Library
Module
Module type
Parameter
Class
Class type

module Input_name : sig ... end

type _1d

type _2d

type _3d

type !'a shape =

type 'a t

val shape : 'a t -> 'a shape

val input : shape:'a shape -> 'a t

val named_input : shape:'a shape -> Input_name.t * 'a t

val dense : _1d t -> shape:int -> _1d t

val conv2d : 
  _3d t ->
  filter:(int * int) ->
  out_channels:int ->
  strides:(int * int) ->
  padding:[ `same | `valid ] ->
  _3d 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 max_pool : 
  _3d t ->
  ksize:(int * int * int * int) ->
  strides:(int * int * int * int) ->
  padding:[ `same | `valid ] ->
  _3d t

val concat : _1d t -> _1d t -> _1d t

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

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

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

val f : float -> shape:'a shape -> 'a t

module Shared_var : sig ... end

module Model : sig ... end