TensorFlow 2.0 Beta is available Learn more

tf.contrib.framework.load_variable

View source on GitHub

Returns a Tensor with the contents of the given variable in the checkpoint.

tf.contrib.framework.load_variable(
    checkpoint_dir,
    name
)

Args:

  • checkpoint_dir: Directory with checkpoints file or path to checkpoint.
  • name: Name of the tensor to return.

Returns:

Tensor object.