جریان تنسور:: عملیات:: GatherV2

برش ها را از axis محور params بر اساس 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 را ببینید.

استدلال ها:

• scope: یک شی Scope
• پارامترها: تانسوری که از آن مقادیر جمع آوری می شود. باید حداقل axis + 1 باشد.
• شاخص ها: تانسور شاخص. باید در محدوده [0, params.shape[axis]) باشد.
• 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
)