SeqModel<InputType, OutputType> constructor

SeqModel<InputType, OutputType>(
  1. List<TapeLayer> layers,
  2. GPUTensor<InputType> input, {
  3. GPUTensor<OutputType>? target,
  4. GPUTensor<Scalar> lossFunction(
    1. GPUTensor<OutputType>,
    2. GPUTensor<OutputType>,
    3. CommandBuffer, {
    4. GPUTensor<Scalar>? outTensor,
    })?,
  5. OptimizerGPU optimizerBuilder(
    1. List<GPUTensor>
    )?,
})

Initializes the Sequential Model with a list of layers and the training input tensor. Optionally accepts a target tensor, lossFunction, and an optimizerBuilder for training workflows.

Implementation

SeqModel(this.layers, this.input,
    {this.target, this.lossFunction, this.optimizerBuilder});