Conozca lo último en aprendizaje automático, IA generativa y más en el
Simposio WiML 2023.
XlaRecvTPUEmbeddingActivations
Organiza tus páginas con colecciones
Guarda y categoriza el contenido según tus preferencias.
Una operación que recibe activaciones de incrustación en la TPU.
El sistema de TPU realiza las búsquedas y agregaciones incrustadas. Los resultados de estas agregaciones son visibles para Tensorflow Graph como las salidas de una operación XlaRecvTPUEmbeddingActivations. Esta operación devuelve una lista que contiene un tensor de activaciones por tabla especificada en el modelo.
Métodos públicos
activaciones estáticas XlaRecvTPUEmbedding | crear ( Ámbito de alcance, Operando <?> deduplicationData, Long numTables, String config) Método de fábrica para crear una clase que envuelve una nueva operación XlaRecvTPUEmbeddingActivations. |
Iterador< Operando <Flotante>> | |
Lista< Salida <Flotante>> | salidas () Una TensorList de activaciones de incrustación que contiene un Tensor por tabla de incrustación en el modelo. |
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 () |
Desde la interfaz java.lang.Iterable vacío | forEach (Consumidor<? super T> arg0) |
Iterador abstracto< Operando <Flotante>> | iterador () |
Spliterator< Operando <Flotante>> | divisor () |
Métodos públicos
Public static XlaRecvTPUEmbeddingActivations create ( Scope scope, Operand <?> deduplicationData, Long numTables, String config)
Método de fábrica para crear una clase que envuelve una nueva operación XlaRecvTPUEmbeddingActivations.
Parámetros
alcance | alcance actual |
---|
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. |
---|
numTablas | El número de tensores de activación de salida. Si el descriptor de características está presente en la configuración de incrustación de tpu, es igual a la cantidad de características, de lo contrario, es igual a la cantidad de tablas de incrustación en el modelo. |
---|
configuración | Proto de configuración de integración de TPU serializado. |
---|
Devoluciones
- una nueva instancia de XlaRecvTPUEmbeddingActivations
public Iterator< Operando <Flotante>> iterador ()
public List< Output <Float>> salidas ()
Una TensorList de activaciones de incrustación que contiene un Tensor por tabla de incrustación en el modelo.
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
}