Trạng thái tối ưu hóa

public struct OptimizerState

Trạng thái toàn cầu được truy cập thông qua StateAccessor .

• init(_:stateCount:)
  Tuyên ngôn

  public init(_ zeros: [Tensor<Float>], stateCount: Int)

• init(đang sao chép:đến:)
  Tuyên ngôn

  public init(copying other: OptimizerState, to device: Device)

• tình trạng
  Tuyên ngôn

  public var state: [Tensor<Float>]

• sải bước
  Tuyên ngôn

  public var stride: Int

• chỉ số dưới(_:_:)
  Tuyên ngôn

  public subscript(stateId: Int, weightId: Int) -> Tensor<Float> { get set }

• chỉ số dưới(_:_:)
  Tuyên ngôn

  public subscript(state: OptimizerWeightStepState, index: StateAccessor) -> Tensor<Float> { get set }