LoadAndRemapMatrix

публичный финальный класс LoadAndRemapMatrix

Загружает двумерный (матричный) «Тензор» с именем «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`, тогда должно выполняться следующее условие:
`(r * num_cols) + (c * num_rows) - (r * c) == len(initializing_values)`

Тензоры переотображения можно сгенерировать с помощью операции 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.

Постоянное значение: «LoadAndRemapMatrix».

Публичные методы

публичный вывод <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.