classe final pública CollectiveAllToAllV2
Troca 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 | ColetivoAllToAllV2.Options | Atributos opcionais para CollectiveAllToAllV2 | |
Métodos Públicos
| Saída <T> | asOutput () Retorna o identificador simbólico de um tensor. |
| Coletivo estáticoAllToAllV2.Options | CommunicationHint (String CommunicationHint) |
| estático <T estende número> CollectiveAllToAllV2 <T> | |
| Saída <T> | dados () |
| Coletivo estáticoAllToAllV2.Options | isStateless (booleano isStateless) |
| Coletivo estáticoAllToAllV2.Options | 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 CollectiveAllToAllV2 <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 CollectiveAllToAllV2.
Parâmetros
| escopo | escopo atual |
|---|---|
| opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de CollectiveAllToAllV2