classe final pública XlaSparseDenseMatmul
Métodos Públicos
Saída <flutuante> | ativações () |
XlaSparseDenseMatmul estático | create (Escopo do escopo , Operando <Integer> rowIds, Operando <?> colIds, Operando <Float> valores, Operando <?> deslocamentos, Operando <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize) Método de fábrica para criar uma classe que envolve uma nova operação XlaSparseDenseMatmul. |
Saída <Inteiro> | rowPointers () |
Saída <Inteiro> | |
Saída <flutuante> | |
Saída <Inteiro> |
Métodos herdados
Métodos Públicos
public static XlaSparseDenseMatmul create ( Escopo de escopo , Operando <Integer> rowIds, Operando <?> colIds, Operando <Float> valores, Operando <?> deslocamentos, Operando <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)
Método de fábrica para criar uma classe que envolve uma nova operação XlaSparseDenseMatmul.
Parâmetros
escopo | escopo atual |
---|
Devoluções
- uma nova instância de XlaSparseDenseMatmul