Gather

برش‌هایی را از محور «پارام‌ها» با توجه به «شاخص‌ها» جمع‌آوری کنید.

«شاخص‌ها» باید یک تانسور صحیح از هر بعد (معمولاً 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 مراجعه کنید.