Загружает двумерный (матричный) «Тензор» с именем «old_tensor_name» из контрольной точки.
в `ckpt_path` и потенциально переупорядочивает его строки и столбцы, используя указанные переназначения.
Большинству пользователей следует использовать один из инициализаторов оболочки (например, tf.contrib.framework.load_and_remap_matrix_initializer) вместо этой функции напрямую.
Переотображения представляют собой одномерные тензоры со следующими свойствами:
- В `row_remapping` должно быть ровно `num_rows` записей. Строка «i» выходной матрицы будет инициализирована из строки, соответствующей индексу «row_remapping[i]» в старом «Тензоре» из контрольной точки.
- В `col_remapping` должно быть либо 0 записей (что указывает на то, что переупорядочение столбцов не требуется), либо записей `num_cols`. Если указано, столбец `j` выходной матрицы будет инициализирован из столбца, соответствующего индексу `col_remapping[j]` в старом `Tensor` из контрольной точки.
- Значение -1 в любом из переназначений означает «отсутствующую» запись. В этом случае значения из тензора `initializing_values` будут использоваться для заполнения недостающей строки или столбца. Если в `row_remapping` есть отсутствующие записи `r`, а в `col_remapping` есть отсутствующие записи `c`, тогда должно выполняться следующее условие:
Тензоры переотображения можно сгенерировать с помощью операции GenerateVocabRemapping.
Например, row_remapping = [1, 0, -1], col_remapping = [0, 2, -1], Initializing_values = [0,5, -0,5, 0,25, -0,25, 42] и w(i, j) представляющее значение из строки i, столбца j старого тензора в контрольной точке, выходная матрица будет выглядеть следующим образом:
[[w(1, 0), w(1, 2), 0,5], [w(0, 0), w(0, 2), -0,5], [0,25, -0,25, 42]]
Вложенные классы
сорт | LoadAndRemapMatrix.Options | Дополнительные атрибуты для LoadAndRemapMatrix |
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Вывод <TFloat32> | какВывод () Возвращает символический дескриптор тензора. |
статический LoadAndRemapMatrix | create ( Область видимости , Операнд < TString > ckptPath, Операнд < TString > oldTensorName, Операнд < TInt64 > rowRemapping, Операнд < TInt64 > colRemapping, Операнд < TFloat32 > инициализацияValues, Длинные numRows, Длинные numCols, Опции... опции) Фабричный метод для создания класса, обертывающего новую операцию LoadAndRemapMatrix. |
статический LoadAndRemapMatrix.Options | maxRowsInMemory (длинный maxRowsInMemory) |
Вывод <TFloat32> | выходная матрица () Выходная матрица, содержащая существующие значения, загруженные из контрольной точки, и все недостающие значения, заполненные из Initializing_values. |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <TFloat32> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static LoadAndRemapMatrix create (область области действия , операнд < TString > ckptPath, операнд < TString > oldTensorName, операнд < TInt64 > rowRemapping, операнд < TInt64 > colRemapping, операнд < TFloat32 > инициализирующие значения, длинные numRows, длинные numCols, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию LoadAndRemapMatrix.
Параметры
объем | текущий объем |
---|---|
ckptPath | Путь к контрольной точке TensorFlow (версия 2, TensorBundle), из которой будет загружена старая матрица Tensor. |
староеTensorName | Имя двумерного «Тензора», загружаемого из контрольной точки. |
rowRemapping | Int `Tensor` переназначений строк (обычно создается с помощью `generate_vocab_remapping`). Даже если перераспределение строк не требуется, это все равно должен быть тензор с индексным значением (например, [0, 1, 2, ...]) или сдвинутый «Тензор» со смещенным индексным значением (например, [8, 9, 10, ...], для секционированных `переменных`). |
colRemapping | Int `Tensor` переназначений столбцов (обычно создается с помощью `generate_vocab_remapping`). Может быть «Тензором» размера 0, если необходимо выполнить только переназначение строк (например, порядок столбцов тот же). |
инициализация значений | Тензор с плавающей запятой, содержащий значения для заполнения ячеек выходной матрицы, которые не загружены из контрольной точки. Длина должна быть точно такой же, как количество недостающих/новых ячеек. |
число строк | Количество строк (длина первого измерения) в выходной матрице. |
numCols | Количество столбцов (длина второго измерения) в выходной матрице. |
параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр LoadAndRemapMatrix
общедоступный статический LoadAndRemapMatrix.Options maxRowsInMemory (Long maxRowsInMemory)
Параметры
МаксРовсинМемори | Максимальное количество строк для одновременной загрузки из контрольной точки. Если меньше или равно 0, вся матрица будет загружена в память. Установка этого аргумента меняет увеличение чтения с диска на меньшее использование памяти. |
---|
публичный вывод <TFloat32> outputMatrix ()
Выходная матрица, содержащая существующие значения, загруженные из контрольной точки, и все недостающие значения, заполненные из Initializing_values.