API reference
-
Builders
-
Models
-
Layers
- Dense
- Activation
- Flatten
- ExpandDims
- Concatenate
- Embedding
- RepeatVector
- Gather
- Max
- Dropout
- BatchNormalization
- Conv1D
- Conv2D
- Conv3D
- MaxPooling1D
- MaxPooling2D
- MaxPooling3D
- AveragePooling1D
- AveragePooling2D
- AveragePooling3D
- GlobalMaxPooling1D
- GlobalMaxPooling2D
- GlobalMaxPooling3D
- GlobalAveragePooling1D
- GlobalAveragePooling2D
- GlobalAveragePooling3D
- SimpleRNN
- LSTM
- GRU
- Attention
- MultiHeadAttention
- Add
- EinsumDense
- InheritMask
-
Activations
-
Losses
-
Optimizers
-
Gradient Tools
-
Gradient Functions
- abs
- add
- bandpart
- cast
- clipByValue
- concat
- div
- equal
- exp
- expandDims
- gather
- get
- greater
- increment
- l2norm
- less
- log
- log1p
- logSoftmax
- masking
- matmul
- maximum
- minimum
- mul
- notEqual
- ones
- onesLike
- randomCategorical
- randomNormal
- reduceMax
- reduceMean
- reduceSum
- relu
- repeat
- reshape
- scale
- shape
- sigmoid
- softmax
- squeeze
- sqrt
- square
- stopGradient
- sub
- tanh
- transpose
- where
- zeros
- zerosLike
-
Datasets
-
Data\Dataset
-
Data\Sequence
-
Data\Image