Метод load

Метод load класса Variable загружает новое значение в переменную. Первым параметром метод принимает значение, которое нужно присвоить переменной. Вторым параметром можно передать сессию, в рамках которой выполняется загрузка. Метод возвращает операцию, которая выполняет загрузку значения.

Метод load отличается от метода assign тем, что возвращает операцию, а не саму переменную. Это удобно, когда нужно выполнить загрузку в рамках сессии TensorFlow.

Синтаксис

variable.load(value, [session])

Пример

Давайте создадим переменную и загрузим в неё новое значение:

import tensorflow as tf

v = tf.Variable([1, 2, 3])
op = v.load([4, 5, 6])
v.assign(op)
print(v)

Результат выполнения кода:

tf.Tensor([4 5 6], shape=(3,), dtype=int32)

Пример

Давайте создадим переменную с типом float32 и загрузим в неё значение:

import tensorflow as tf

v = tf.Variable([1.0, 2.0, 3.0], dtype=tf.float32)
op = v.load([4.0, 5.0, 6.0])
v.assign(op)
print(v)

Результат выполнения кода:

tf.Tensor([4. 5. 6.], shape=(3,), dtype=float32)

Пример

Давайте создадим двумерную переменную и загрузим в неё новое значение:

import tensorflow as tf

v = tf.Variable([[1, 2, 3], [4, 5, 6]])
op = v.load([[7, 8, 9], [10, 11, 12]])
v.assign(op)
print(v)

Результат выполнения кода:

tf.Tensor(
[[ 7  8  9]
 [10 11 12]], shape=(2, 3), dtype=int32)

Смотрите также

класс Variable,
который создает переменную TensorFlow
метод assign,
который присваивает переменной новое значение
метод read_value,
который читает значение переменной
метод numpy,
который возвращает значение переменной как массив NumPy