tensorflow::ops::BarrierTakeMany

Retira de uma barreira o número fornecido de elementos concluídos.

Resumo

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

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

Argumentos:

• escopo: Um objeto de escopo
• maçaneta: A maçaneta de uma barreira.
• num_elements: Um tensor de um único elemento contendo o número de elementos a serem considerados.
• component_types: O tipo de cada componente em um valor.

Atributos opcionais (ver Attrs ):

• allow_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 ficará bloqueada por até timeout_ms milissegundos. Observação: esta opção ainda não é suportada.

Devoluções:

• Índices Output : Um tensor unidimensional de índices, com comprimento num_elems. Esses índices referem-se ao lote no qual os valores foram inseridos 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 de qualquer dimensão por componente em um elemento de barreira. Todos os valores têm comprimento num_elements na dimensão 0.

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
)