tf.train.load_variable

TensorFlow 2 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.