パブリック静的クラスEnqueueTPUEmbeddingBatch.Options
EnqueueTPUEmbeddingBatch
のオプションの属性
パブリックメソッド
EnqueueTPUEmbeddingBatch.Options | コンバイナー(List<String> コンバイナー) |
EnqueueTPUEmbeddingBatch.Options | deviceOrdinal (長い deviceOrdinal) |
継承されたメソッド
パブリックメソッド
public EnqueueTPUEmbeddingBatch.Optionsコンバイナー(List<String> コンバイナー)
パラメーター
結合器 | 文字列スカラーのリスト。重み付き合計後の埋め込みアクティブ化を正規化する方法を指定する各埋め込みテーブルに 1 つあります。サポートされている結合子は、「mean」、「sum」、または「sqrtn」です。 'mean' の重みの合計が 0 になること、または 'sqrtn' の 2 乗重みの合計が 0 になることは無効です。コンバイナーが渡されない場合、デフォルトではすべてのテーブルに「sum」が使用されます。 |
---|
public EnqueueTPUEmbeddingBatch.Options deviceOrdinal (Long deviceOrdinal)
パラメーター
デバイス通常 | 使用する TPU デバイス。これは、Op が TPU デバイスで実行されている場合は -1 である必要があり、Op が CPU デバイスで実行されている場合は >= 0 である必要があります。 |
---|