XlaSparseDenseMatmul

XlaSparseDenseMatmul คลาสสุดท้ายสาธารณะ

วิธีการสาธารณะ

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

วิธีการสืบทอด

วิธีการสาธารณะ

การเปิด ใช้ งานเอาต์พุต สาธารณะ <Float> ()

สร้าง XlaSparseDenseMatmul สาธารณะแบบคงที่ (ขอบเขต ขอบเขต , ตัวดำเนินการ <Integer> rowIds, ตัวดำเนินการ <?> colIds, ตัว ดำเนินการ <Float> ค่า, ตัวดำเนินการ <?> ออฟเซ็ต, ตัวดำเนินการ <Float> embeddingTable, ยาว maxIdsPerPartition, ยาว maxUniqueIdsPerPartition, ยาว inputSize)

วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ XlaSparseDenseMatmul ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
การส่งคืน
  • อินสแตนซ์ใหม่ของ XlaSparseDenseMatmul

สาธารณะ เอาท์พุท <จำนวนเต็ม> rowPointers ()

สาธารณะ เอาท์พุท <จำนวนเต็ม> sortedEmbeddingIds ()

เอาท์พุท สาธารณะ <ลอย> sortedGains ()

สาธารณะ เอาท์พุท <จำนวนเต็ม> sortedSampleIds ()