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

#include <array_ops.h>

รวบรวม ชิ้นส่วนจาก params ตาม indices

สรุป

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

    # Scalar indices
    output[:, ..., :] = params[indices, :, ... :]

    # Vector indices
    output[i, :, ..., :] = params[indices[i], :, ... :]

    # Higher rank indices
    output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]

หาก indices เป็นการเรียงสับเปลี่ยนและ len(indices) == params.shape[0] การดำเนินการนี้จะสับเปลี่ยน params ตามลำดับ

validate_indices : เลิกใช้แล้ว หากการดำเนินการนี้ถูกกำหนดให้กับ CPU ค่าใน indices จะถูกตรวจสอบให้อยู่ภายในช่วงเสมอ หากกำหนดให้กับ GPU ดัชนีนอกขอบเขตจะส่งผลให้เกิดการทำงานที่ปลอดภัยแต่ไม่ได้ระบุ ซึ่งอาจรวมถึงการทำให้เกิดข้อผิดพลาด

ข้อโต้แย้ง:

ผลตอบแทน:

  • Output : เทนเซอร์เอาท์พุต

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

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

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

operation
output

งานสาธารณะ

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

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

ValidateIndices (bool x)

โครงสร้าง

เทนเซอร์โฟลว์ :: ops :: รวบรวม :: Attrs

ตัวตั้งค่าคุณลักษณะเพิ่มเติมสำหรับ Gather

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

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

Operation operation

เอาท์พุท

::tensorflow::Output output

งานสาธารณะ

รวมตัว

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

รวมตัว

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

โหนด

::tensorflow::Node * node() const 

ตัวดำเนินการ::tensorflow::อินพุต

 operator::tensorflow::Input() const 

ตัวดำเนินการ::tensorflow::เอาต์พุต

 operator::tensorflow::Output() const 

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

ตรวจสอบดัชนี

Attrs ValidateIndices(
  bool x
)