public struct OptimizerState
StateAccessor
के माध्यम से वैश्विक स्थिति तक पहुँचा जा सकता है।
घोषणा
public init(_ zeros: [Tensor<Float>], stateCount: Int)
घोषणा
public init(copying other: OptimizerState, to device: Device)
घोषणा
public var state: [Tensor<Float>]
घोषणा
public var stride: Int
घोषणा
public subscript(stateId: Int, weightId: Int) -> Tensor<Float> { get set }
घोषणा
public subscript(state: OptimizerWeightStepState, index: StateAccessor) -> Tensor<Float> { get set }