공개 최종 클래스 XlaSparseDenseMatmul
공개 방법
출력 <플로트> | 활성화 () |
정적 XlaSparseDenseMatmul | |
출력 <정수> | 행 포인터 () |
출력 <정수> | |
출력 <플로트> | sortedGains () |
출력 <정수> |
상속된 메서드
공개 방법
public static XlaSparseDenseMatmul create ( 범위 범위, 피연산자 <Integer> rowIds, 피연산자 <?> colIds, 피연산자 <Float> 값, 피연산자 <?> 오프셋, 피연산자 <Float> embeddingTable, Long maxIdsPerPartition, Long maxUniqueIdsPerPartition, Long inputSize)
새로운 XlaSparseDenseMatmul 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
범위 | 현재 범위 |
---|
보고
- XlaSparseDenseMatmul의 새 인스턴스