テンソルフロー::作戦::リソースSparseApplyMomentum
コレクションでコンテンツを整理
必要に応じて、コンテンツの保存と分類を行います。
#include <training_ops.h>
モメンタムスキームに従って、「*var」と「*accum」の関連エントリを更新します。
まとめ
Nesterov の勢いを使用したい場合は、use_nesterov = True を設定します。
つまり、grad がある行については、次のように var と accum を更新します。
accum = accum * 勢い + grad var -= lr * accum
引数:
- スコープ:スコープオブジェクト
- var: Variable() から取得する必要があります。
- accum: Variable() から取得する必要があります。
- lr: 学習率。スカラーでなければなりません。
- grad: グラデーション。
- indices: var と accum の最初の次元へのインデックスのベクトル。
- 勢い:勢い。スカラーでなければなりません。
オプションの属性 ( Attrs
を参照):
- use_locking:
True
の場合、var テンソルと accum テンソルの更新はロックによって保護されます。それ以外の場合、動作は未定義ですが、競合が少なくなる可能性があります。 - use_nesterov:
True
の場合、compute grad に渡されるテンソルは var - lr * momento * accum となるため、最終的に取得する var は実際には var - lr *勢い * accum になります。
戻り値:
- 作成された
Operation
コンストラクターとデストラクター | |
---|---|
ResourceSparseApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum) | |
ResourceSparseApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum, const ResourceSparseApplyMomentum::Attrs & attrs) |
公共機能 | |
---|---|
operator::tensorflow::Operation () const |
パブリック静的関数 | |
---|---|
UseLocking (bool x) | |
UseNesterov (bool x) |
パブリック属性
手術
Operation operation
公共機能
リソースSparseApplyMomentum
ResourceSparseApplyMomentum(
const ::tensorflow::Scope & scope,
::tensorflow::Input var,
::tensorflow::Input accum,
::tensorflow::Input lr,
::tensorflow::Input grad,
::tensorflow::Input indices,
::tensorflow::Input momentum
)
リソースSparseApplyMomentum
ResourceSparseApplyMomentum(
const ::tensorflow::Scope & scope,
::tensorflow::Input var,
::tensorflow::Input accum,
::tensorflow::Input lr,
::tensorflow::Input grad,
::tensorflow::Input indices,
::tensorflow::Input momentum,
const ResourceSparseApplyMomentum::Attrs & attrs
)
演算子::tensorflow::オペレーション
operator::tensorflow::Operation() const
パブリック静的関数
ロックを使用する
Attrs UseLocking(
bool x
)
ネステロフを使用する
Attrs UseNesterov(
bool x
)
テンソルフロー::作戦::リソースSparseApplyMomentum
#include <training_ops.h>
モメンタムスキームに従って、「*var」と「*accum」の関連エントリを更新します。
まとめ
Nesterov の勢いを使用したい場合は、use_nesterov = True を設定します。
つまり、grad がある行については、次のように var と accum を更新します。
accum = accum * 勢い + grad var -= lr * accum
引数:
- スコープ:スコープオブジェクト
- var: Variable() から取得する必要があります。
- accum: Variable() から取得する必要があります。
- lr: 学習率。スカラーでなければなりません。
- grad: グラデーション。
- indices: var と accum の最初の次元へのインデックスのベクトル。
- 勢い:勢い。スカラーでなければなりません。
オプションの属性 ( Attrs
を参照):
- use_locking:
True
の場合、var テンソルと accum テンソルの更新はロックによって保護されます。それ以外の場合、動作は未定義ですが、競合が少なくなる可能性があります。 - use_nesterov:
True
の場合、compute grad に渡されるテンソルは var - lr * momento * accum となるため、最終的に取得する var は実際には var - lr *勢い * accum になります。
戻り値:
- 作成された
Operation
コンストラクターとデストラクター | |
---|---|
ResourceSparseApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum) | |
ResourceSparseApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum, const ResourceSparseApplyMomentum::Attrs & attrs) |
公共機能 | |
---|---|
operator::tensorflow::Operation () const |
パブリック静的関数 | |
---|---|
UseLocking (bool x) | |
UseNesterov (bool x) |
パブリック属性
手術
Operation operation
公共機能
リソースSparseApplyMomentum
ResourceSparseApplyMomentum(
const ::tensorflow::Scope & scope,
::tensorflow::Input var,
::tensorflow::Input accum,
::tensorflow::Input lr,
::tensorflow::Input grad,
::tensorflow::Input indices,
::tensorflow::Input momentum
)
リソースSparseApplyMomentum
ResourceSparseApplyMomentum(
const ::tensorflow::Scope & scope,
::tensorflow::Input var,
::tensorflow::Input accum,
::tensorflow::Input lr,
::tensorflow::Input grad,
::tensorflow::Input indices,
::tensorflow::Input momentum,
const ResourceSparseApplyMomentum::Attrs & attrs
)