EnqueueTPUEmbeddingSparseBatch

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

op ที่เข้าคิวดัชนีอินพุต TPUEmbedding จาก SparseTensor

Op นี้ช่วยลดความยุ่งยากในการย้ายรหัสที่ใช้ embedding_lookup_sparse() แม้ว่าการประมวลผลล่วงหน้าของ Python บางส่วนของอาร์กิวเมนต์ SparseTensor เพื่อ embedding_lookup_sparse() จะต้องสร้างอาร์กิวเมนต์สำหรับ Op นี้ เนื่องจากอนุญาตให้ใช้ EnqueueTPUEmbeddingSparseBatch Op เดียวเท่านั้นต่อหนึ่งขั้นตอนการฝึกอบรม

เทนเซอร์ที่ตำแหน่งที่สอดคล้องกันในรายการอินพุตทั้งสามต้องมีรูปร่างเหมือนกัน นั่นคืออันดับ 1 ที่มี dim_size() เท่ากับจำนวนการค้นหาทั้งหมดในตารางที่อธิบายโดย table_id ที่เกี่ยวข้อง

คลาสที่ซ้อนกัน

ระดับ

EnqueueTPUEการฝังSparseBatch.Options

แอตทริบิวต์เสริมสำหรับ EnqueueTPUEmbeddingSparseBatch

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

คงที่ EnqueueTPUEmbeddingSparseBatch.Options	ตัวรวม (รายการ <String> ตัวรวม)
คงที่ <T ขยายจำนวน U ขยายจำนวน V ขยายจำนวน> EnqueueTPUEmbeddingSparseBatch	สร้าง ( ขอบเขต ขอบเขต Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ EnqueueTPUEmbeddingSparseBatch ใหม่
คงที่ EnqueueTPUEmbeddingSparseBatch.Options	deviceOrdinal (อุปกรณ์แบบยาวOrdinal)

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

จากคลาส org.tensorflow.op.PrimitiveOp

บูลีนสุดท้าย	เท่ากับ (วัตถุ obj)
intสุดท้าย	แฮชโค้ด ()
การดำเนินการ	ออฟ () ส่งคืน Operation พื้นฐาน
สตริงสุดท้าย	toString ()

จากคลาส java.lang.Object

บูลีน	เท่ากับ (วัตถุ arg0)
คลาสสุดท้าย<?>	getClass ()
int	แฮชโค้ด ()
สุดท้ายเป็นโมฆะ	แจ้ง ()
สุดท้ายเป็นโมฆะ	แจ้งทั้งหมด ()
สตริง	toString ()
สุดท้ายเป็นโมฆะ	รอ (arg ยาว, int arg1)
สุดท้ายเป็นโมฆะ	รอ (ยาว arg0)
สุดท้ายเป็นโมฆะ	รอ ()

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

EnqueueTPUEmbeddingSparseBatch.Options แบบคงที่สาธารณะ (List <String> combiners )

พารามิเตอร์

เครื่องผสม	รายการของสเกลาร์สตริง หนึ่งรายการสำหรับแต่ละตารางการฝังที่ระบุวิธีการทำให้การเปิดใช้งานการฝังเป็นปกติหลังจากการรวมแบบถ่วงน้ำหนัก ตัวรวมที่รองรับคือ 'mean', 'sum' หรือ 'sqrtn' ไม่ถูกต้องที่จะมีผลรวมของน้ำหนักเป็น 0 สำหรับ 'ค่าเฉลี่ย' หรือผลรวมของน้ำหนักยกกำลังสองเป็น 0 สำหรับ 'sqrtn' หากไม่ผ่านตัวรวม ค่าเริ่มต้นคือการใช้ 'ผลรวม' สำหรับทุกตาราง

EnqueueTPUEmbeddingSparseBatch สาธารณะแบบคงที่ สร้าง ( ขอบเขต ขอบเขต, Iterable< Operand <T>> sampleIndices, Iterable< Operand <U>> embeddingIndices, Iterable< Operand <V>> aggregationWeights, Operand <String> modeOverride, Options... ตัวเลือก)

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

พารามิเตอร์

ขอบเขต	ขอบเขตปัจจุบัน
ดัชนีตัวอย่าง	รายการเทนเซอร์อันดับ 1 ที่ระบุตัวอย่างการฝึกและคุณลักษณะที่มีค่า embedding_indices และ aggregation_weights ที่สอดคล้องกัน sample_indices[i] ต้องเท่ากับ b * nf + f โดยที่ nf คือจำนวนของคุณสมบัติจากตารางที่เกี่ยวข้อง f อยู่ใน [0, nf) และ b อยู่ใน [0, ขนาดแบทช์)
การฝังดัชนี	รายการเทนเซอร์อันดับ 1 ดัชนีลงในตารางการฝัง
การรวมน้ำหนัก	รายการเทนเซอร์อันดับ 1 ที่มีต่อตัวอย่าง - เช่น ต่อ (ตัวอย่างการฝึกอบรม คุณลักษณะ) - น้ำหนักรวม
modeOverride	อินพุตสตริงที่แทนที่โหมดที่ระบุใน TPUEmbeddingConfiguration ค่าที่รองรับคือ {'unspecified', 'inference', 'training', 'backward_pass_only'} เมื่อตั้งค่าเป็น 'ไม่ระบุ' โหมดที่ตั้งค่าใน TPUEmbeddingConfiguration จะถูกใช้ มิฉะนั้น mode_override จะถูกใช้
ตัวเลือก	ดำเนินการค่าแอตทริบิวต์ทางเลือก

คืนสินค้า

• อินสแตนซ์ใหม่ของ EnqueueTPUEmbeddingSparseBatch

EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal แบบ คงที่สาธารณะ (Long deviceOrdinal)

พารามิเตอร์

อุปกรณ์ลำดับ	อุปกรณ์ TPU ที่จะใช้ ควรเป็น >= 0 และน้อยกว่าจำนวนคอร์ TPU ในงานที่วางโหนด