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