Gather

ชั้นเรียนสุดท้ายสาธารณะ รวบรวม

รวบรวมชิ้นส่วนจากแกน 'params' 'axis' ตาม 'ดัชนี'

`ดัชนี` จะต้องเป็นเทนเซอร์จำนวนเต็มของมิติใดๆ (โดยปกติจะเป็น 0-D หรือ 1-D) สร้างเอาท์พุตเทนเซอร์ที่มีรูปร่าง `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]` โดยที่:

# Scalar indices (output is rank(params) - 1).
     output[a_0, ..., a_n, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices, b_0, ..., b_n]
 
     # Vector indices (output is rank(params)).
     output[a_0, ..., a_n, i, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
 
     # Higher rank indices (output is rank(params) + rank(indices) - 1).
     output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
       params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
 

โปรดทราบว่าบน CPU หากพบดัชนีนอกขอบเขต ข้อผิดพลาดจะถูกส่งกลับ บน GPU หากพบดัชนีนอกขอบเขต 0 จะถูกเก็บไว้ในค่าเอาต์พุตที่สอดคล้องกัน

ดูเพิ่มเติมที่ `tf.batch_gather` และ tf.gather_nd

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

ระดับ รวบรวมตัวเลือก คุณลักษณะทางเลือกสำหรับ Gather

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

เอาท์พุต <T>
เป็นเอาท์พุต ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
Gather.Options แบบคงที่
ชุด Dims (ชุดยาว)
คงที่ <T, U ขยายหมายเลข, V ขยายหมายเลข> รวบรวม <T>
สร้าง (ขอบเขต ขอบเขต , พารามิเตอร์ ตัวดำเนินการ <T>, ดัชนี ตัวถูกดำเนินการ <U>, แกนตัว ถูกดำเนินการ <V>, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวบรวมการดำเนินการรวบรวมใหม่
เอาท์พุต <T>
เอาท์พุท ()
ค่าจาก `params` ที่รวบรวมจากดัชนีที่กำหนดโดย `indices` โดยมีรูปร่าง `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`

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

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

เอาท์ พุท สาธารณะ <T> asOutput ()

ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์

อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต

สาธารณะคง Gather.Options ชุด Dims (ชุดยาว Dims)

สาธารณะคง ที่ สร้าง <T> (ขอบเขต ขอบเขต , พารามิเตอร์ ตัวดำเนินการ <T>, ดัชนีตัว ดำเนินการ <U>, แกนตัว ถูกดำเนินการ <V>, ตัวเลือก... ตัวเลือก)

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

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
พารามิเตอร์ เทนเซอร์ที่ใช้รวบรวมค่าต่างๆ ต้องมีอันดับอย่างน้อย `แกน + 1`
ดัชนี ดัชนีเทนเซอร์ ต้องอยู่ในช่วง `[0, params.shape[axis])`
แกน แกนใน `params` เพื่อรวบรวม `ดัชนี` ค่าเริ่มต้นเป็นมิติแรก รองรับดัชนีเชิงลบ
ตัวเลือก มีค่าแอตทริบิวต์ทางเลือก
การส่งคืน
  • ตัวอย่างใหม่ของ Gather

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

ค่าจาก `params` ที่รวบรวมจากดัชนีที่กำหนดโดย `indices` โดยมีรูปร่าง `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`