TensorFlow張量變換函數(shù)：tf.gather

函數(shù)：tf.gather

gather(
    params,
    indices,
    validate_indices=None,
    name=None,
    axis=0
)

定義在：tensorflow/python/ops/array_ops.py.

參見指南：張量變換>切片和連接

根據(jù)索引從參數(shù)軸上收集切片.
索引必須是任何維度的整數(shù)張量 (通常為 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]

參數(shù)：

• params：一個(gè)張量.這個(gè)張量是用來收集數(shù)值的.該張量的秩必須至少是 axis + 1.
  
• indices：一個(gè)張量.必須是以下類型之一：int32,int64.索引張量必須在 [0, params.shape[axis]) 范圍內(nèi).
• axis：一個(gè)張量.必須是以下類型之一：int32,int64.在參數(shù)軸從中收集索引.默認(rèn)為第一個(gè)維度.支持負(fù)索引.
• name：操作的名稱(可選).

返回值：

該函數(shù)返回一個(gè)張量.與參數(shù)具有相同的類型.參數(shù)值從索引給定的索引中收集而來,并且形狀為：params.shape[:axis] + indices.shape + params.shape[axis + 1:].