tensorflow:: ops:: GatherV2

Recopilar segmentos del axis de params según los indices .

Resumen

indices deben ser un tensor entero de cualquier dimensión (normalmente 0-D o ​​1-D). Produce un tensor de salida con forma params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:] donde:

    # 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]

Tenga en cuenta que, en la CPU, si se encuentra un índice fuera de rango, se devuelve un error. En la GPU, si se encuentra un índice fuera de rango, se almacena un 0 en el valor de salida correspondiente.

Véase también tf.batch_gather y tf.gather_nd .

Argumentos:

• ámbito: Un objeto de ámbito
• params: El tensor del que se obtendrán los valores. Debe tener al menos rango axis + 1 .
• índices: Tensor de índices. Debe estar en el rango [0, params.shape[axis]) .
• eje: El eje en params del que se obtendrán indices . Por defecto, se utiliza la primera dimensión. Admite índices negativos.

Devoluciones:

• Output : Valores de params recopilados de los índices dados por indices , con forma params.shape[:axis] + indices.shape + params.shape[axis + 1:] .

Atributos públicos

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
)