OptimizerWeightStepState

public struct OptimizerWeightStepState

Estado para uma única etapa de um único peso dentro de um otimizador.

  • Hiperparâmetros.

    Declaração

    public let globals: [Tensor<Float>]
  • Valores temporários (só podem ser atribuídos uma vez).

    Declaração

    public var locals: [Tensor<Float>]
  • A derivada real do peso em relação à função de perda.

    Declaração

    public var grad: Tensor<Float>
  • O peso sendo otimizado.

    Declaração

    public let weight: Tensor<Float>
  • A saída final do otimizador. (deve ser definido apenas uma vez). nil significa que o peso não será tocado. Isto irá ser aplicado para o verdadeiro peso no final: weight += step .

    Declaração

    public var step: Tensor<Float>?
  • Declaração

    public subscript(local: LocalAccessor) -> Tensor<Float> { get set }
  • Declaração

    public subscript(global: GlobalAccessor) -> Tensor<Float> { get }