パブリック最終クラスXlaSparseDenseMatmul
パブリックメソッド
出力<浮動小数点> | |
静的XlaSparseDenseMatmul | |
出力<整数> | rowPointers () |
出力<整数> | |
出力<浮動小数点数> | ソートゲイン() |
出力<整数> |
継承されたメソッド
パブリックメソッド
public static XlaSparseDenseMatmul create (スコープスコープ、オペランド<Integer> rowIds、オペランド<?>colIds、オペランド<Float>値、オペランド<?>オフセット、オペランド<Float>embeddingTable、Long maxIdsPerPartition、Long maxUniqueIdsPerPartition、Long inputSize)
新しい XlaSparseDenseMatmul 操作をラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|
戻り値
- XlaSparseDenseMatmul の新しいインスタンス