Conozca lo último en aprendizaje automático, IA generativa y más en el
Simposio WiML 2023.
XlaSendTPUEmbeddingGradients
Organiza tus páginas con colecciones
Guarda y categoriza el contenido según tus preferencias.
Una operación que realiza actualizaciones de gradiente de tablas de incrustación.
El argumento de gradientes es una TensorList que tiene la misma longitud y formas que el valor de retorno de XlaRecvTPUEmbeddingActivations, pero contiene gradientes de la pérdida del modelo con respecto a las activaciones de incrustación. Las tablas de incrustación se actualizan a partir de estos gradientes a través del optimizador especificado en el protocolo TPUEmbeddingConfiguration proporcionado a tpu.initialize_system.
Métodos públicos
estático XlaSendTPUEmbeddingGradients | crear ( Ámbito de alcance, Iterable< Operando <Flotante>> gradientes, Iterable< Operando <Flotante>> tasas de aprendizaje, Operando <?> deduplicationData, Configuración de cadena) Método de fábrica para crear una clase que envuelve una nueva operación XlaSendTPUEmbeddingGradients. |
Métodos Heredados
De la clase java.lang.Object booleano | es igual a (Objeto arg0) |
clase final<?> | obtenerClase () |
En t | código hash () |
vacío final | notificar () |
vacío final | notificar a todos () |
Cuerda | a la cadena () |
vacío final | espera (largo arg0, int arg1) |
vacío final | espera (largo arg0) |
vacío final | espera () |
Métodos públicos
XlaSendTPUEmbeddingGradients estáticos públicos crear ( Ámbito de alcance, Iterable< Operando <Flotante>> gradientes, Iterable< Operando <Flotante>> tasas de aprendizaje, Operando <?> datos de deduplicación, Configuración de cadena)
Método de fábrica para crear una clase que envuelve una nueva operación XlaSendTPUEmbeddingGradients.
Parámetros
alcance | alcance actual |
---|
gradientes | Una TensorList de degradados con los que actualizar las tablas de incrustación. |
---|
tasas de aprendizaje | Una TensorList de tasas de aprendizaje utilizadas para actualizar las tablas de incrustación a través del optimizador. La longitud de TensorList debe ser igual a la cantidad de etiquetas de índice de aprendizaje dinámico especificadas en el protocolo TPUEmbeddingConfiguration. |
---|
datos de deduplicación | Un tensor con type=DT_VARIANT que contiene los datos de deduplicación. El tensor es una tupla anidada XLA que contiene N elementos (donde N es la proporción del número de incrustaciones a núcleos tensoriales por chip de TPU). Cada elemento de la tupla anidada es una tupla de tensores de rango 1. Cada tensor contiene índices (DT_UINT32) para incrustar la búsqueda en TensorCore o pesos (DT_FLOAT) para aplicar a la salida de la operación de incrustación de búsqueda. |
---|
configuración | Proto de configuración de integración de TPU serializado. |
---|
Devoluciones
- una nueva instancia de XlaSendTPUEmbeddingGradients
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. Cierto contenido está sujeto a la licencia de NumPy.
Última actualización: 2023-10-06 (UTC)
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Falta la información que necesito"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Muy complicado o demasiados pasos"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Desactualizado"
},{
"type": "thumb-down",
"id": "translationIssue",
"label":"Problema de traducción"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Problema con las muestras o los códigos"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Otro"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Fácil de comprender"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Resolvió mi problema"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Otro"
}]
{
"lastModified": "Última actualización: 2023-10-06 (UTC)",
"confidential": False
}