Aviso: Este projeto está obsoleto. O Swift for TensorFlow foi um experimento na plataforma de próxima geração para aprendizado de máquina, incorporando as pesquisas mais recentes sobre aprendizado de máquina, compiladores, programação diferenciável, design de sistemas e muito mais. Foi arquivado em fevereiro de 2021.

Esta página foi traduzida pela API Cloud Translation.

Estado do Otimizador

public struct OptimizerState

Estado global acessado por meio de StateAccessor .

init(_:stateCount:)

Declaração

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

init(copiando:para:)

Declaração

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

estado
Declaração
public var state: [Tensor<Float>]
passo
Declaração
public var stride: Int

subscrito(_:_:)

Declaração

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

subscrito(_:_:)

Declaração

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