EnqueueTPUEmbeddingSparseBatch.Options

パブリック静的クラスEnqueueTPUEmbeddingSparseBatch.Options

EnqueueTPUEmbeddingSparseBatchのオプション属性

パブリックメソッド

EnqueueTPUEmbeddingSparseBatch.Options
コンバイナー(List<String> コンバイナー)
EnqueueTPUEmbeddingSparseBatch.Options
deviceOrdinal (長い deviceOrdinal)

継承されたメソッド

パブリックメソッド

public EnqueueTPUEmbeddingSparseBatch.Optionsコンバイナー(List<String> コンバイナー)

パラメーター
結合器文字列スカラーのリスト。重み付き合計後の埋め込みアクティブ化を正規化する方法を指定する各埋め込みテーブルに 1 つあります。サポートされている結合子は、「mean」、「sum」、または「sqrtn」です。 'mean' の重みの合計が 0 になること、または 'sqrtn' の 2 乗重みの合計が 0 になることは無効です。コンバイナーが渡されない場合、デフォルトではすべてのテーブルに「sum」が使用されます。

public EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)

パラメーター
デバイス通常使用する TPU デバイス。 0 以上で、ノードが配置されているタスク内の TPU コアの数未満である必要があります。