lớp cuối cùng công khai XlaSparseDenseMatmulWithCsrInput
Phương pháp công khai
| Đầu ra <Float> | kích hoạt () |
| Đầu ra <Float> | asOutput () Trả về phần điều khiển tượng trưng của một tenxơ. |
| tĩnh XlaSparseDenseMatmulWithCsrInput | tạo ( Phạm vi phạm vi , Toán hạng <Integer> rowPointers, Toán hạng <Integer> đã sắp xếpSampleIds, Toán hạng <Integer> đã sắp xếpTokenIds, Toán hạng <Float> đã sắp xếpGains, Toán hạng <Float> embeddingTable, Toán hạng <Integer> numMinibatchesPerPhysicalSparseCore, Kích thước đầu vào dài, Lượng tử hóa nổiConfigLow, Lượng tử hóa nổiConfigHigh, Lượng tử hóa dàiConfigNumBuckets, Chuỗi tên bảng) Phương thức xuất xưởng để tạo một lớp bao bọc thao tác XlaSparseDenseMatmulWithCsrInput mới. |
Phương pháp kế thừa
Phương pháp công khai
Đầu ra công khai <Float> asOutput ()
Trả về phần điều khiển tượng trưng của một tenxơ.
Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.
public static XlaSparseDenseMatmulWithCsrInput create ( Phạm vi phạm vi , Toán hạng <Integer> rowPointers, Toán hạng <Integer> đã sắp xếpSampleIds, Toán hạng <Integer> được sắp xếpTokenIds, Toán hạng <Float> đã sắp xếpGains, Toán hạng <Float> embeddingTable, Toán hạng <Integer> numMinibatchesPerPhysicalSparseCore, Kích thước đầu vào dài, Lượng tử hóa floatConfigLow, Lượng tử hóa floatConfigHigh, Lượng tử hóa dàiConfigNumBuckets, String tableName)
Phương thức xuất xưởng để tạo một lớp bao bọc thao tác XlaSparseDenseMatmulWithCsrInput 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 XlaSparseDenseMatmulWithCsrInput