XlaSparseDenseMatmul คลาสสุดท้ายสาธารณะ
วิธีการสาธารณะ
| เอาท์พุต <โฟลต> | |
| XlaSparseDenseMatmul แบบคงที่ | สร้าง (ขอบเขต ขอบเขต , ตัวดำเนินการ <Integer> rowIds, ตัวดำเนินการ <?> colIds, ค่าตัว ดำเนิน การ <Float>, ตัว ดำเนิน การ <?> ออฟเซ็ต, ตัวดำเนินการ <Float> embeddingTable, ยาว maxIdsPerPartition, ยาว maxUniqueIdsPerPartition, ยาว inputSize) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ XlaSparseDenseMatmul ใหม่ |
| เอาต์พุต <จำนวนเต็ม> | ตัวชี้แถว () |
| เอาต์พุต <จำนวนเต็ม> | |
| เอาท์พุต <โฟลต> | |
| เอาต์พุต <จำนวนเต็ม> |
วิธีการสืบทอด
วิธีการสาธารณะ
สร้าง XlaSparseDenseMatmul สาธารณะแบบคงที่ (ขอบเขต ขอบเขต , ตัวดำเนินการ <Integer> rowIds, ตัวดำเนินการ <?> colIds, ตัว ดำเนินการ <Float> ค่า, ตัวดำเนินการ <?> ออฟเซ็ต, ตัวดำเนินการ <Float> embeddingTable, ยาว maxIdsPerPartition, ยาว maxUniqueIdsPerPartition, ยาว inputSize)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ XlaSparseDenseMatmul ใหม่
พารามิเตอร์
| ขอบเขต | ขอบเขตปัจจุบัน |
|---|
การส่งคืน
- อินสแตนซ์ใหม่ของ XlaSparseDenseMatmul