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 usando las reasignaciones especificadas.
La mayoría de los usuarios deberían usar uno de los inicializadores de envoltura (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á desde la fila correspondiente al índice` row_remapping [i] `en el antiguo` Tensor` del 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á desde la columna correspondiente al índice` col_remapping [j] `en el antiguo` Tensor` del punto de control.
- Un valor de -1 en cualquiera de las reasignaciones significa una entrada "faltante". En ese caso, los valores del tensor `initializing_values` se usarán para completar la fila o columna que falta. Si a `row_remapping` le faltan` r` entradas y a `col_remapping` le faltan` c`, la siguiente condición debe ser verdadera:
Los tensores de reasignación se pueden generar utilizando la opción GenerateVocabRemapping.
Por ejemplo, con row_remapping = [1, 0, -1], col_remapping = [0, 2, -1], initializing_values = [0.5, -0.5, 0.25, -0.25, 42] yw (i, j) representando 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.Options | Los atributos opcionales para LoadAndRemapMatrix |
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
Salida < TFloat32 > | asOutput () Devuelve el identificador simbólico del tensor. |
estática LoadAndRemapMatrix | crear ( Alcance alcance, operando < TString > ckptPath, operando < TString > oldTensorName, operando < TInt64 > rowRemapping, operando < TInt64 > colRemapping, operando < TFloat32 > initializingValues, Long numRows, númeroColumnas largas, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación LoadAndRemapMatrix. |
estáticas LoadAndRemapMatrix.Options | maxRowsInMemory (Long maxRowsInMemory) |
Salida < TFloat32 > | outputMatrix () Matriz de salida que contiene los valores existentes cargados desde el punto de control y con los valores faltantes rellenados desde initializing_values. |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Métodos públicos
pública de salida < 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.
public static LoadAndRemapMatrix crear ( Alcance alcance, operando < TString > ckptPath, operando < TString > oldTensorName, operando < TInt64 > rowRemapping, operando < TInt64 > colRemapping, operando < TFloat32 > initializingValues, Long numRows, númeroColumnas largas, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación LoadAndRemapMatrix.
Parámetros
alcance | alcance actual |
---|---|
ckptPath | Ruta al punto de control de TensorFlow (versión 2, `TensorBundle`) desde el cual se cargará la matriz antigua` Tensor`. |
oldTensorName | Nombre del `Tensor` 2-D que se cargará desde el punto de control. |
rowRemapping | Un `Tensor` int de reasignaciones de filas (generalmente creado por` generate_vocab_remapping`). Incluso si no se necesita reasignación de filas, debe ser un tensor con valores de índice (p. Ej., [0, 1, 2, ...]) o un `Tensor` con valores de índice desplazados (p. Ej., [8, 9, 10, ...], para `Variables` particionadas). |
colRemapping | 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). |
initializingValues | Un "Tensor" flotante que contiene valores para completar en 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. |
numRows | Número de filas (longitud de la 1ª dimensión) en la matriz de salida. |
numCols | Número de columnas (longitud de la 2ª dimensión) en la matriz de salida. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de LoadAndRemapMatrix
public static LoadAndRemapMatrix.Options maxRowsInMemory (Long maxRowsInMemory)
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 que 0, toda la matriz se cargará en la memoria. Al configurar este argumento, se intercambian más lecturas de disco por un menor uso de memoria. |
---|