przepływ tensorowy:: ops:: Zebrać

Zbierz plasterki z params według indices .

Streszczenie

indices muszą być tensorami całkowitymi o dowolnym wymiarze (zwykle 0-D lub 1-D). Tworzy tensor wyjściowy z indices.shape + params.shape[1:] gdzie:

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

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

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

Jeśli indices są permutacją i len(indices) == params.shape[0] wówczas ta operacja odpowiednio permutuje params .

validate_indices : PRZESTARZAŁE. Jeśli ta operacja jest przypisana do procesora, wartości w indices są zawsze sprawdzane, czy mieszczą się w dopuszczalnym zakresie. Jeśli są przypisane do procesora graficznego, indeksy poza zakresem powodują bezpieczne, ale nieokreślone zachowanie, które może obejmować zgłaszanie błędu.

Argumenty:

• zakres: Obiekt Scope

Zwroty:

• Output : Tensor wyjściowy.

Atrybuty publiczne

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
)