Spec-Zone.ru › TensorFlow

tf.raw_ops.LoadAndRemapMatrix

Загружает двумерный (матрицу) Tensor с именем old_tensor_name из контрольной точки

Псевдонимы

Псевдонимы для совместимости при миграции

Дополнительные сведения см. в руководстве по миграции.

tf.compat.v1.raw_ops.LoadAndRemapMatrix

tf.raw_ops.LoadAndRemapMatrix(
    ckpt_path,
    old_tensor_name,
    row_remapping,
    col_remapping,
    initializing_values,
    num_rows,
    num_cols,
    max_rows_in_memory=-1,
    name=None
)

в ckpt_path и при необходимости переупорядочивает его строки и столбцы с помощью указанных схем переназначения.

Большинству пользователей следует использовать одну из обёрточных функций-инициализаторов (например, tf.contrib.framework.load_and_remap_matrix_initializer), а не вызывать эту функцию напрямую.

Схемы переназначения — это одномерные тензоры со следующими свойствами:

  • row_remapping должен содержать ровно num_rows элементов. Строка i выходной матрицы будет инициализирована строкой с индексом row_remapping[i] в старом Tensor из контрольной точки.
  • 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]]

Аргументы
ckpt_path Tensor типа string. Путь к контрольной точке TensorFlow (версия 2, TensorBundle), из которой будет загружена старая матрица Tensor.
old_tensor_name Tensor типа string. Имя двумерного Tensor, которое нужно загрузить из контрольной точки.
row_remapping Tensor типа int64. Целочисленный Tensor переназначений строк (обычно создаётся с помощью generate_vocab_remapping). Даже если переназначение строк не требуется, это всё равно должен быть тензор со значениями-индексами (например, [0, 1, 2, ...]) или тензор со значениями-индексами со смещением Tensor (например, [8, 9, 10, ...] для разделённых на части Variables).
col_remapping Tensor типа int64. Целочисленный Tensor переназначений столбцов (обычно создаётся с помощью generate_vocab_remapping). Может быть тензором Tensor с размером 0, если требуется переназначить только строки (например, когда порядок столбцов не меняется).
initializing_values Tensor типа float32. Тензор с плавающей точкой Tensor, содержащий значения для ячеек выходной матрицы, которые не загружаются из контрольной точки. Длина должна точно совпадать с количеством отсутствующих / новых ячеек.
num_rows int со значением >= 0. Количество строк (длина первого измерения) выходной матрицы.
num_cols int со значением >= 1. Количество столбцов (длина второго измерения) выходной матрицы.
max_rows_in_memory Необязательный int. По умолчанию — -1. Максимальное количество строк, загружаемых из контрольной точки за один раз. Если значение меньше или равно 0, вся матрица будет загружена в память. Этот аргумент позволяет снизить потребление памяти ценой увеличения количества операций чтения с диска.
name Имя операции (необязательно).
Возвращает
Tensor типа float32.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/LoadAndRemapMatrix

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API