tensorflow :: operaciones :: Recolectar

Reúna porciones de params acuerdo con los indices .

Resumen

indices deben ser un tensor entero de cualquier dimensión (generalmente 0-D o ​​1-D). Produce un tensor de salida con shape 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] , esta operación permutará los params consecuencia.

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

Argumentos:

• alcance: un objeto de alcance

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
)