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