tensoreflusso:: ops:: ResourceApplyMomentum

bookmark_border Mantieni tutto organizzato con le raccolte Salva e classifica i contenuti in base alle tue preferenze.

Aggiorna '*var' secondo lo schema del momentum.

Riepilogo

Imposta use_nesterov = True se desideri utilizzare lo slancio di Nesterov.

accum = accum * momento + grad var -= lr * accum

Argomenti:

• scope: un oggetto Scope
• var: dovrebbe provenire da una variabile().
• accum: dovrebbe provenire da una variabile().
• lr: fattore di scala. Deve essere uno scalare.
• grad: il gradiente.
• slancio: slancio. Deve essere uno scalare.

Attributi facoltativi (vedi Attrs ):

• use_locking: Se True , l'aggiornamento dei tensori var e accum sarà protetto da un lock; altrimenti il ​​comportamento non è definito, ma può mostrare meno contesa.
• use_nesterov: Se True , il tensore passato al calcolo grad sarà var - lr * momentum * accum, quindi alla fine, il var che ottieni è in realtà var - lr * momentum * accum.

Resi:

• l' Operation creata

Attributi pubblici

Operation operation

 ResourceApplyMomentum(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input accum,
  ::tensorflow::Input lr,
  ::tensorflow::Input grad,
  ::tensorflow::Input momentum
)

 ResourceApplyMomentum(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input accum,
  ::tensorflow::Input lr,
  ::tensorflow::Input grad,
  ::tensorflow::Input momentum,
  const ResourceApplyMomentum::Attrs & attrs
)

 operator::tensorflow::Operation() const 

Attrs UseLocking(
  bool x
)

Attrs UseNesterov(
  bool x
)

Operation operation

 ResourceApplyMomentum(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input accum,
  ::tensorflow::Input lr,
  ::tensorflow::Input grad,
  ::tensorflow::Input momentum
)

 ResourceApplyMomentum(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input accum,
  ::tensorflow::Input lr,
  ::tensorflow::Input grad,
  ::tensorflow::Input momentum,
  const ResourceApplyMomentum::Attrs & attrs
)

 operator::tensorflow::Operation() const 

Attrs UseLocking(
  bool x
)

Attrs UseNesterov(
  bool x
)