tensorflow:: operaciones:: ReunirV2

Reúna cortes del eje del axis de params de acuerdo con los indices .

Resumen

los indices deben ser un tensor entero de cualquier dimensión (generalmente 0-D o ​​1-D). Produce un tensor de salida con forma params.shape[:axis] + indices.shape + 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 límite, se devuelve un error. En GPU, si se encuentra un índice fuera de límites, se almacena un 0 en el valor de salida correspondiente.

Consulte también tf.batch_gather y tf.gather_nd .

Argumentos:

• alcance: un objeto de alcance
• params: el tensor del que se obtienen valores. Debe ser al menos axis + 1 .
• índices: Índice tensor. Debe estar en el rango [0, params.shape[axis]) .
• eje: el eje en params para recopilar indices . El valor predeterminado es la primera dimensión. Admite índices negativos.

Devoluciones:

• Output : valores de params recopilados de í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
)