The following classes are available globally.

  • A mutable, shareable, owning reference to a tensor.


    public final class Parameter<Scalar> where Scalar : TensorFlowScalar
  • Class wrapping a C pointer to a TensorHandle. This class owns the TensorHandle and is responsible for destroying it.


    public class TFETensorHandle : _AnyTensorHandle
  • SGD

    Stochastic gradient descent (SGD) optimizer.

    An optimizer that implements stochastic gradient descent, with support for momentum, learning rate decay, and Nesterov momentum.


    public class SGD<Model: Differentiable>: Optimizer
        where Model.TangentVector: VectorProtocol & ElementaryFunctions,
              Model.TangentVector.VectorSpaceScalar == Float