XlaSparseDenseMatmul

lớp cuối cùng công khai XlaSparseDenseMatmul

Phương pháp công cộng

Đầu ra <Float>
tĩnh XlaSparseDenseMatmul
tạo ( Phạm vi phạm vi, Toán hạng <Integer> rowIds, Toán hạng <?> colIds, Giá trị toán hạng <Float>, Độ lệch toán hạng <?>, Bảng nhúng toán hạng <Float>, MaxIdsPerPartition dài, MaxUniqueIdsPerPartition dài, Kích thước đầu vào dài)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác XlaSparseDenseMatmul mới.
Đầu ra <Số nguyên>
Đầu ra <Số nguyên>
Đầu ra <Float>
Đầu ra <Số nguyên>

Phương pháp kế thừa

Phương pháp công cộng

Kích hoạt đầu ra công khai <Float> ()

công khai tĩnh XlaSparseDenseMatmul tạo ( Phạm vi phạm vi , Toán hạng <Integer> rowIds, Toán hạng <?> colIds, Giá trị toán hạng <Float>, Độ lệch toán hạng <?>, Bảng nhúng toán hạng <Float>, MaxIdsPerPartition dài, MaxUniqueIdsPerPartition dài, Kích thước đầu vào dài)

Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác XlaSparseDenseMatmul mới.

Thông số
phạm vi phạm vi hiện tại
Trả lại
  • một phiên bản mới của XlaSparseDenseMatmul

Đầu ra công khai <Integer> rowPointers ()

Đầu ra công khai <Integer> được sắp xếpEmbeddingIds ()

Đầu ra công khai <Float> được sắp xếpGains ()

Đầu ra công khai <Integer> được sắp xếpSampleIds ()