تينسورفلو :: العمليات :: يجتمع

اجمع الشرائح من 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 : موتر الإخراج.

السمات العامة

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 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs ValidateIndices(
  bool x
)