Carga un `Tensor` 2-D (matriz) con el nombre `old_tensor_name` desde el punto de control
en `ckpt_path` y potencialmente reordena sus filas y columnas utilizando las reasignaciones especificadas.
La mayoría de los usuarios deberían usar uno de los inicializadores de contenedor (como `tf.contrib.framework.load_and_remap_matrix_initializer`) en lugar de esta función directamente.
Las reasignaciones son tensores 1-D con las siguientes propiedades:
- `row_remapping` debe tener exactamente entradas `num_rows`. La fila `i` de la matriz de salida se inicializará a partir de la fila correspondiente al índice `row_remapping[i]` en el antiguo `Tensor` desde el punto de control.
- `col_remapping` debe tener 0 entradas (lo que indica que no es necesario reordenar las columnas) o entradas `num_cols`. Si se especifica, la columna `j` de la matriz de salida se inicializará a partir de la columna correspondiente al índice `col_remapping[j]` en el antiguo `Tensor` desde el punto de control.
- Un valor de -1 en cualquiera de las reasignaciones significa que falta una entrada. En ese caso, los valores del tensor `initializing_values` se utilizarán para completar la fila o columna que falta. Si a `row_remapping` le faltan entradas `r` y a `col_remapping` le faltan entradas `c`, entonces la siguiente condición debe ser verdadera:
Los tensores de reasignación se pueden generar usando la operación GenerateVocabRemapping.
Como ejemplo, con row_remapping = [1, 0, -1], col_remapping = [0, 2, -1], initializing_values = [0.5, -0.5, 0.25, -0.25, 42] y w(i, j) Al representar el valor de la fila i, columna j del antiguo tensor en el punto de control, la matriz de salida se verá así:
[[w(1, 0), w(1, 2), 0,5], [w(0, 0), w(0, 2), -0,5], [0,25, -0,25, 42]]
Clases anidadas
clase | LoadAndRemapMatrix.Opciones | Atributos opcionales para LoadAndRemapMatrix |
Constantes
Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
Salida <TFloat32> | como salida () Devuelve el identificador simbólico del tensor. |
LoadAndRemapMatrix estático | crear (alcance alcance , Operando < TString > ckptPath, Operando < TString > oldTensorName, Operando < TInt64 > rowRemapping, Operando < TInt64 > colRemapping, Operando < TFloat32 > inicializandoValores, Long numRows, Long numCols, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación LoadAndRemapMatrix. |
LoadAndRemapMatrix.Options estático | maxRowsInMemory (maxRowsInMemory largo) |
Salida <TFloat32> | matriz de salida () Matriz de salida que contiene los valores existentes cargados desde el punto de control y con los valores faltantes completados desde valores_inicializados. |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
Salida pública <TFloat32> 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.
creación pública estática de LoadAndRemapMatrix (alcance del alcance , operando <TString> ckptPath, operando <TString> oldTensorName, operando <TInt64> rowRemapping, operando <TInt64> colRemapping, operando <TFloat32> inicializandoValues, Long numRows, Long numCols, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación LoadAndRemapMatrix.
Parámetros
alcance | alcance actual |
---|---|
ckptRuta | Ruta al punto de control de TensorFlow (versión 2, `TensorBundle`) desde donde se cargará la antigua matriz `Tensor`. |
antiguoNombreTensor | Nombre del `Tensor` 2-D que se cargará desde el punto de control. |
filaRemapeo | Un `Tensor` int de reasignaciones de filas (generalmente creado por `generate_vocab_remapping`). Incluso si no es necesaria la reasignación de filas, debe ser un tensor con valor de índice (por ejemplo, [0, 1, 2, ...]) o un "tensor" con valor de índice desplazado (por ejemplo, [8, 9, 10, ...], para `Variables` particionadas). |
colRemapeo | Un `Tensor` int de reasignaciones de columnas (generalmente creado por `generate_vocab_remapping`). Puede ser un `Tensor` de tamaño 0 si solo se va a realizar la reasignación de filas (por ejemplo, el orden de las columnas es el mismo). |
inicializandoValores | Un `Tensor` flotante que contiene valores para completar las celdas de la matriz de salida que no se cargan desde el punto de control. La longitud debe ser exactamente la misma que la cantidad de celdas nuevas o faltantes. |
número de filas | Número de filas (longitud de la primera dimensión) en la matriz de salida. |
numCols | Número de columnas (longitud de la segunda dimensión) en la matriz de salida. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de LoadAndRemapMatrix
LoadAndRemapMatrix.Options estático público maxRowsInMemory (maxRowsInMemory largo)
Parámetros
maxRowsInMemory | El número máximo de filas para cargar desde el punto de control a la vez. Si es menor o igual a 0, se cargará toda la matriz en la memoria. Configurar este argumento intercambia mayores lecturas de disco por un menor uso de memoria. |
---|
Salida pública <TFloat32> salidaMatrix ()
Matriz de salida que contiene los valores existentes cargados desde el punto de control y con los valores faltantes completados desde valores_inicializados.