XlaSparseDenseMatmul

공개 최종 클래스 XlaSparseDenseMatmul

공개 방법

출력 <플로트>
정적 XlaSparseDenseMatmul
생성 ( 범위 범위, 피연산자 <Integer> rowIds, 피연산자 <?> colIds, 피연산자 <Float> 값, 피연산자 <?> 오프셋, 피연산자 <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)
새로운 XlaSparseDenseMatmul 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <정수>
출력 <정수>
출력 <플로트>
출력 <정수>

상속된 메서드

공개 방법

공개 출력 <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 ()