tensorflow:: ops:: Gather

Recopilar segmentos de params según 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 indices.shape + params.shape[1:] donde:

    # Scalar indices
    output[:, ..., :] = params[indices, :, ... :]

    # Vector indices
    output[i, :, ..., :] = params[indices[i], :, ... :]

    # Higher rank indices
    output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]

Si indices es una permutación y len(indices) == params.shape[0] entonces esta operación permutará params en consecuencia.

validate_indices : OBSOLETO. Si esta operación se asigna a la CPU, los valores de indices siempre se validan para que estén dentro del rango. Si se asigna a la GPU, los índices fuera de rango dan como resultado un comportamiento seguro pero no especificado, que puede incluir la generación de un error.

Argumentos:

• ámbito: Un objeto de ámbito

Devoluciones:

• Output : El tensor de salida.

Atributos públicos

Operation operation

::tensorflow::Output output

 Gather(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices
)

 Gather(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  const Gather::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs ValidateIndices(
  bool x
)