เทนเซอร์โฟลว์::ออปส์::แกเธอร์วี2

รวบรวม ส่วนต่างๆ จาก axis params ตาม indices

สรุป

indices ต้องเป็นเทนเซอร์จำนวนเต็มที่มีมิติใดก็ได้ (โดยปกติจะเป็น 0 มิติหรือ 1 มิติ) ฟังก์ชันนี้จะสร้างเทนเซอร์เอาต์พุตที่มีรูปร่างเป็น params.shape[:axis] + indices.shape + 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

ข้อโต้แย้ง:

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

การส่งคืนสินค้า:

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

คุณลักษณะสาธารณะ

Operation operation

::tensorflow::Output output

 GatherV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  ::tensorflow::Input axis
)

 GatherV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  ::tensorflow::Input axis,
  const GatherV2::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs BatchDims(
  int64 x
)