public struct OptimizerStateGlobal state accessed through StateAccessor.
-
Declaration
public init(_ zeros: [Tensor<Float>], stateCount: Int) -
Declaration
public init(copying other: OptimizerState, to device: Device) -
Declaration
public var state: [Tensor<Float>] -
Declaration
public var stride: Int -
Declaration
public subscript(stateId: Int, weightId: Int) -> Tensor<Float> { get set } -
Declaration
public subscript(state: OptimizerWeightStepState, index: StateAccessor) -> Tensor<Float> { get set }