Conozca lo último en aprendizaje automático, IA generativa y más en el
Simposio WiML 2023.
Google uses AI technology to translate content into your preferred language. AI translations can contain errors.
CrossReplicaSum
Organiza tus páginas con colecciones
Guarda y categoriza el contenido según tus preferencias.
Una operación para sumar entradas en instancias de TPU replicadas.
Cada instancia proporciona su propia entrada.
Por ejemplo, suponga que hay 8 instancias de TPU: "[A, B, C, D, E, F, G, H]". Pasar group_assignment = `[[0,2,4,6], [1,3,5,7]]` establece `A, C, E, G` como grupo 0 y` B, D, F, H` como grupo 1. Así obtenemos las salidas: `[A + C + E + G, B + D + F + H, A + C + E + G, B + D + F + H, A + C + E + G, B + D + F + H, A + C + E + G, B + D + F + H] `.
Constantes
| Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos heredados
De la clase java.lang.Object | booleano | equals (Object arg0) |
| Clase final <?> | getClass () |
| En t | hashCode () |
| vacío final | notificar () |
| vacío final | notifyAll () |
| Cuerda | toString () |
| vacío final | espera (tiempo arg0, arg1 int) |
| vacío final | espera (arg0 largo) |
| vacío final | wait () |
Desde la interfaz org.tensorflow.Operand | abstracta de salida <T> | asOutput () Devuelve el identificador simbólico del tensor. |
| T abstracto | asTensor () Devuelve el tensor en este operando. |
| abstracta Forma | forma () Devuelve el (posiblemente parcialmente conocida) forma del tensor se hace referencia por la Output de esta operando. |
| clase abstracta <T> | escriba () Devuelve el tipo de tensor de este operando |
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Valor constante: "CrossReplicaSum"
Métodos públicos
pública de salida <T> asOutput ()
Devuelve el identificador simbólico del 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.
Método de fábrica para crear una clase que envuelva una nueva operación CrossReplicaSum.
Parámetros
| alcance | alcance actual |
|---|
| aporte | La entrada local a la suma. |
|---|
| asignación grupal | Un tensor int32 con forma [num_groups, num_replicas_per_group]. `group_assignment [i]` representa los ID de réplica en el i-ésimo subgrupo. |
|---|
Devoluciones
- una nueva instancia de CrossReplicaSum
pública de salida <T> salida ()
La suma de todas las entradas distribuidas.
Salvo que se indique lo contrario, el contenido de esta página está sujeto a la licencia Atribución 4.0 de Creative Commons, y los ejemplos de código están sujetos a la licencia Apache 2.0. Para obtener más información, consulta las políticas del sitio de Google Developers. Java es una marca registrada de Oracle o sus afiliados.
Última actualización: 2021-11-29 (UTC)
[[["Fácil de comprender","easyToUnderstand","thumb-up"],["Resolvió mi problema","solvedMyProblem","thumb-up"],["Otro","otherUp","thumb-up"]],[["Falta la información que necesito","missingTheInformationINeed","thumb-down"],["Muy complicado o demasiados pasos","tooComplicatedTooManySteps","thumb-down"],["Desactualizado","outOfDate","thumb-down"],["Problema de traducción","translationIssue","thumb-down"],["Problema con las muestras o los códigos","samplesCodeIssue","thumb-down"],["Otro","otherDown","thumb-down"]],["Última actualización: 2021-11-29 (UTC)"],[],[]]
