public struct OptimizerState Estado global acessado através StateAccessor .
Declaração
public init(_ zeros: [Tensor<Float>], stateCount: Int)Declaração
public init(copying other: OptimizerState, to device: Device)Declaração
public var state: [Tensor<Float>]Declaração
public var stride: IntDeclaração
public subscript(stateId: Int, weightId: Int) -> Tensor<Float> { get set }Declaração
public subscript(state: OptimizerWeightStepState, index: StateAccessor) -> Tensor<Float> { get set }