Intercambia mutuamente múltiples tensores de idéntico tipo y forma.
`is_stateless` significa que cada operación no necesita dependencias de control con otras operaciones colectivas. En este caso, se deben utilizar claves que sean únicas en tiempo de ejecución (por ejemplo, `instance_key`) para distinguir los grupos colectivos.
Clases anidadas
clase | ColectivoAllToAllV2.Options | Atributos opcionales para CollectiveAllToAllV2 |
Métodos públicos
Salida <T> | como salida () Devuelve el identificador simbólico de un tensor. |
CollectiveAllToAllV2.Options estático | sugerencia de comunicación (sugerencia de comunicación de cadena) |
estático <T extiende Número> CollectiveAllToAllV2 <T> | crear (ámbito de alcance , entrada de operando <T>, tamaño de grupo de operando <entero>, clave de grupo de operando <entero>, clave de instancia de operando <entero>, iterable < operando <?>> token de pedido, opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación CollectiveAllToAllV2. |
Salida <T> | datos () |
CollectiveAllToAllV2.Options estático | isStateless (booleano isStateless) |
CollectiveAllToAllV2.Options estático | timeoutSeconds (Tiempo de espera flotante) |
Métodos heredados
Métodos públicos
Salida pública <T> como Salida ()
Devuelve el identificador simbólico de un tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
public static CollectiveAllToAllV2 <T> create (Ámbito de alcance , Operando <T> entrada, Operando <Integer> tamaño de grupo, Operando <Integer> clave de grupo, Operando <Integer> clave de instancia, Iterable < Operando <?>> orderingToken, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación CollectiveAllToAllV2.
Parámetros
alcance | alcance actual |
---|---|
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de CollectiveAllToAllV2