tf.train.load_variable | TensorFlow v2.16.1 (original) (raw)

tf.train.load_variable

Stay organized with collections Save and categorize content based on your preferences.

Returns the tensor value of the given variable in the checkpoint.

View aliases

Compat aliases for migration

tf.compat.v1.train.load_variable

tf.train.load_variable(
    ckpt_dir_or_file, name
)

When the variable name is unknown, you can use tf.train.list_variables to inspect all the variable names.

Example usage:

import tensorflow as tf
a = tf.Variable(1.0)
b = tf.Variable(2.0)
ckpt = tf.train.Checkpoint(var_list={'a': a, 'b': b})
ckpt_path = ckpt.save('tmp-ckpt')
var= tf.train.load_variable(
    ckpt_path, 'var_list/a/.ATTRIBUTES/VARIABLE_VALUE')
print(var)  # 1.0

Args
ckpt_dir_or_file	Directory with checkpoints file or path to checkpoint.
name	Name of the variable to return.

Returns
A numpy ndarray with a copy of the value of this variable.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.