ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.train.load_variable

TensorFlow 1 version View source on GitHub

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

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

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