XlaSparseDenseMatmul

публичный финальный класс XlaSparseDenseMatmul

Публичные методы

Выход <С плавающей запятой>
статический XlaSparseDenseMatmul
create ( Область области действия, Operand <Integer> rowIds, Operand <?> colIds, Operand <Float> значения, Operand <?> смещения, Operand <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)
Фабричный метод для создания класса, обертывающего новую операцию XlaSparseDenseMatmul.
Вывод <Целое число>
Вывод <Целое число>
Выход <С плавающей запятой>
Вывод <Целое число>

Унаследованные методы

Публичные методы

public Output <Float> активации ()

public static XlaSparseDenseMatmul create (область области действия , операнд <Integer> rowIds, операнд <?> colIds, значения операнда <Float>, смещения операнда <?>, операнд <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)

Фабричный метод для создания класса, обертывающего новую операцию XlaSparseDenseMatmul.

Параметры
объем текущий объем
Возврат
  • новый экземпляр XlaSparseDenseMatmul

общедоступный вывод <Integer> rowPointers ()

публичный вывод <Integer> sortedEmbeddingIds ()

публичный вывод <Float> sortedGains ()

публичный вывод <Integer> sortedSampleIds ()