API reference

• Builders

  • NeuralNetworks
  • Models
  • Layers
  • Losses
  • Optimizers
  • Gradient
  • Datasets
  • Data

• Models

  • Sequential
  • ModelLoader
  • AbstractModel

• 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

  • ReLU
  • Sigmoid
  • Softmax
  • Tanh

• Losses

  • MeanSquaredError
  • BinaryCrossEntropy
  • CategoricalCrossEntropy
  • SparseCategoricalCrossEntropy
  • Huber

• Optimizers

  • SGD
  • Adam
  • RMSprop

• Gradient Tools

  • Variable
  • GradientTape
  • GraphFunction
  • ndarray
  • ArraySpec
  • modules

• Gradient Functions

  • add
  • bandpart
  • cast
  • clipByValue
  • div
  • equal
  • exp
  • get
  • greater
  • increment
  • less
  • log
  • matmul
  • mul
  • notEqual
  • ones
  • onesLike
  • reduceMean
  • reduceSum
  • repeat
  • reshape
  • scale
  • shape
  • sqrt
  • square
  • stopGradient
  • sub
  • transpose
  • zeros
  • zerosLike

• Datasets

  • MNIST
  • Fashion-MNIST
  • CIFAR-10

• Data\Dataset

  • NDArrayDataset
  • CSVDataset
  • ClassifiedDirectoryDataset
  • SequentialDataset

• Data\Sequence

  • Preprocessor
  • Tokenizer
  • TextClassifiedDataset

• Data\Image

  • ImageFilter
  • ImageDataGenerator
  • ImageClassifiedDataset