เทนเซอร์โฟลว์:: ปฏิบัติการ:: GatherV2

#include <array_ops.h>

รวบรวม สไลซ์จากแกน params มส์ axis ตาม indices

สรุป

indices ต้องเป็นเทนเซอร์จำนวนเต็มของมิติใดๆ (ปกติคือ 0-D หรือ 1-D) สร้างเทนเซอร์เอาต์พุตที่มีรูปร่าง 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

อาร์กิวเมนต์:

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

ผลตอบแทน:

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

ตัวสร้างและตัวทำลาย

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)

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

operation
output

งานสาธารณะ

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

ฟังก์ชั่นคงที่สาธารณะ

BatchDims (int64 x)

โครงสร้าง

tensorflow:: ops:: GatherV2:: Attrs

ตัวตั้งค่าแอตทริบิวต์เสริมสำหรับ GatherV2

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

การดำเนินการ

Operation operation

ผลผลิต

::tensorflow::Output output

งานสาธารณะ

GatherV2

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

GatherV2

 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 

ฟังก์ชั่นคงที่สาธารณะ

BatchDims

Attrs BatchDims(
  int64 x
)