tensorflow:: عملیات:: GatherV2

برش‌ها را از params axis بر اساس indices جمع‌آوری کنید .

خلاصه

indices باید یک تانسور عدد صحیح با هر بُعدی (معمولاً 0-D یا 1-D) باشند. یک تانسور خروجی با شکل params.shape[:axis] + indices.shape[batch_dims:] + 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 مراجعه کنید.

استدلال‌ها:

• دامنه: یک شیء دامنه
• پارامترها: تانسوری که مقادیر از آن جمع‌آوری می‌شوند. حداقل باید rank axis + 1 باشد.
• شاخص‌ها: تانسور شاخص. باید در محدوده [0, params.shape[axis]) باشد.
• محور: محوری که در params برای جمع‌آوری indices از آن استفاده می‌شود. پیش‌فرض‌ها روی بعد اول است. از اندیس‌های منفی پشتیبانی می‌کند.

بازده:

• Output : مقادیر params که از اندیس‌های داده شده توسط indices جمع‌آوری شده‌اند، با shape 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
)