옵티마이저WeightStepState

public struct OptimizerWeightStepState

옵티마이저 내 단일 가중치의 단일 단계에 대한 상태입니다.

  • 초매개변수.

    선언

    public let globals: [Tensor<Float>]
  • 임시 값(한 번만 할당할 수 있음)

    선언

    public var locals: [Tensor<Float>]
  • 손실 함수에 대한 가중치의 실제 미분입니다.

    선언

    public var grad: Tensor<Float>
  • 무게가 최적화되고 있습니다.

    선언

    public let weight: Tensor<Float>
  • 옵티마이저의 최종 출력입니다. (실제로는 한 번만 설정해야 합니다). nil은 무게가 건드리지 않음을 의미합니다. 이는 마지막에 실제 가중치에 적용됩니다( weight += step .

    선언

    public var step: Tensor<Float>?
  • 선언

    public subscript(local: LocalAccessor) -> Tensor<Float> { get set }
  • 선언

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