Предупреждение: этот проект устарел. Swift для TensorFlow был экспериментом с платформой нового поколения для машинного обучения, включающей в себя последние исследования в области машинного обучения, компиляторов, дифференцируемого программирования, проектирования систем и многого другого. Он был заархивирован в феврале 2021 года.

Эта страница переведена с помощью Cloud Translation API.

Оптимизаторстате

public struct OptimizerState

Доступ к глобальному состоянию осуществляется через StateAccessor .

инициализация(_:stateCount:)
Декларация
public init(_ zeros: [Tensor<Float>], stateCount: Int)

init(копирование:в:)

Декларация

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 }