przepływ tensorowy:: ops:: Zebrać

Zbierz plastry z params zgodnie z indices .

Streszczenie

indices muszą być tensorami całkowitymi o dowolnym wymiarze (zwykle 0-D lub 1-D). Tworzy tensor wyjściowy z indeksami kształtu.shape 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ą, a len(indices) == params.shape[0] to ta operacja spowoduje odpowiednią params .

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

Argumenty:

• zakres: Obiekt zakresu

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
)