| View source on GitHub |
Returns the tensor value of the given variable in the checkpoint.
tf.train.load_variable(
ckpt_dir_or_file, name
)
| 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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/train/load_variable