CollectiveGatherV2

aula final pública CollectiveGatherV2

Acumula mutuamente vários tensores de tipo e formato idênticos.

`is_stateless` significa que cada operação não precisa de dependências de controle para outras operações coletivas. Neste caso, chaves que são únicas em tempo de execução (por exemplo, `instance_key`) devem ser usadas para distinguir grupos coletivos.

Classes aninhadas

aula ColetivoGatherV2.Opções Atributos opcionais para CollectiveGatherV2

Métodos Públicos

Saída <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
CollectiveGatherV2.Options estático
CommunicationHint (String CommunicationHint)
estático <T estende número> CollectiveGatherV2 <T>
create ( Escopo do escopo , Operando <T> entrada, Operando <Integer> groupSize, Operando <Integer> groupKey, Operando <Integer> instanceKey, Iterable< Operando <?>> orderingToken, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação CollectiveGatherV2.
Saída <T>
dados ()
CollectiveGatherV2.Options estático
isStateless (booleano isStateless)
CollectiveGatherV2.Options estático
timeoutSeconds (tempo limite flutuanteSeconds)

Métodos herdados

Métodos Públicos

Saída pública <T> asOutput ()

Retorna o identificador simbólico de um tensor.

As entradas para operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.

public static CollectiveGatherV2.Options CommunicationHint (String CommunicationHint)

public static CollectiveGatherV2 <T> create ( Escopo , Operando <T> entrada, Operando <Integer> groupSize, Operando <Integer> groupKey, Operando <Integer> instanceKey, Iterable< Operando <?>> orderingToken, Opções... opções)

Método de fábrica para criar uma classe que envolve uma nova operação CollectiveGatherV2.

Parâmetros
escopo escopo atual
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância do CollectiveGatherV2

dados de saída pública <T> ()

public static CollectiveGatherV2.Options isStateless (Boolean isStateless)

public static CollectiveGatherV2.Options timeoutSeconds (Float timeoutSeconds)