deepr.utils.get_fetchable_tensors

deepr.utils.get_fetchable_tensors(graph, names)[source]

Retrieve fetch tensors from graph.

Parameters:

graph (tf.Graph) – A Tensorflow Graph
names (List[str]) – List of operations or tensor names

Returns:

Mapping of names to tf.Tensor

Return type:

Dict[str, tf.Tensor]