fluxo tensor:: ops:: BarreiraTakeMany

Pega um determinado número de elementos concluídos de uma barreira.

Resumo

Esta operação concatena tensores de componentes de elementos completos ao longo da 0ª dimensão para formar um tensor de componente único.

Os elementos saem da barreira quando estão completos e na ordem em que foram colocados na barreira. A saída dos índices fornece informações sobre o lote em que cada elemento foi originalmente inserido na barreira.

Argumentos:

• escopo: um objeto Escopo
• alça: A alça de uma barreira.
• num_elements: um tensor de elemento único contendo o número de elementos a serem obtidos.
• component_types: o tipo de cada componente em um valor.

Atributos opcionais (veja Attrs ):

• permitir_small_batch: Permite retornar menos de num_elements itens se a barreira já estiver fechada.
• timeout_ms: Se a fila estiver vazia, esta operação será bloqueada por até timeout_ms milissegundos. Nota: Esta opção ainda não é suportada.

Retorna:

• Índices Output : um tensor unidimensional de índices, com comprimento num_elems. Esses índices referem-se ao lote em que os valores foram colocados na barreira (começando com MIN_LONG e aumentando a cada BarrierInsertMany ).
• Chaves Output : um tensor unidimensional de chaves, com comprimento num_elements.
• Valores OutputList : um tensor qualquer dimensional por componente em um elemento de barreira. Todos os valores têm comprimento num_elements na 0ª dimensão.

Atributos públicos

::tensorflow::Output indices

::tensorflow::Output keys

Operation operation

::tensorflow::OutputList values

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types
)

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types,
  const BarrierTakeMany::Attrs & attrs
)

Attrs AllowSmallBatch(
  bool x
)

Attrs TimeoutMs(
  int64 x
)

Attrs WaitForIncomplete(
  bool x
)