OptimizerState

public struct OptimizerState

Estado global accede a través de StateAccessor .

• init (_: stateCount :)

  Declaración

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

• init (copiando: a :)

  Declaración

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

• estado

  Declaración

  public var state: [Tensor<Float>]

• paso

  Declaración

  public var stride: Int

• subíndice(_:_:)

  Declaración

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

• subíndice(_:_:)

  Declaración

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