XlaSparseDenseMatmul

パブリック最終クラスXlaSparseDenseMatmul

パブリックメソッド

出力<浮動小数点>
アクティベーション()
静的XlaSparseDenseMatmul
create (スコープscope、オペランド<Integer> rowIds、オペランド<?>colIds、オペランド<Float>値、オペランド<?>オフセット、オペランド<Float>embeddingTable、Long maxIdsPerPartition、Long maxUniqueIdsPerPartition、Long inputSize)
新しい XlaSparseDenseMatmul 操作をラップするクラスを作成するファクトリ メソッド。
出力<整数>
rowPointers ()
出力<整数>
sortedEmbeddingIds ()
出力<浮動小数点数>
ソートゲイン()
出力<整数>
sortedSampleIds ()

継承されたメソッド

パブリックメソッド

public Output <Float>アクティベーション()

public static XlaSparseDenseMatmul create (スコープスコープ、オペランド<Integer> rowIds、オペランド<?>colIds、オペランド<Float>値、オペランド<?>オフセット、オペランド<Float>embeddingTable、Long maxIdsPerPartition、Long maxUniqueIdsPerPartition、Long inputSize)

新しい XlaSparseDenseMatmul 操作をラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
戻り値
  • XlaSparseDenseMatmul の新しいインスタンス

public Output <Integer> rowPointers ()

public Output <Integer> selectedEmbeddingIds ()

public出力<Float> selectedGains ()

public Output <Integer> sortedSampleIds ()