INNER CODE UNIT · Python
row_sum
deepchem/deepchem · contrib/tensorflow_models/__init__.py:44
row_sum = np.sum(x, axis=2)
x /= row_sum.reshape(x.shape[:2] + (1,))
return x
class TensorflowGraph(object):
"""Simple class that holds information needed to run Tensorflow graph."""
def __init__(self, graph, session, name_scopes, output, labels, weights,
loss):
warnings.warn(
"TensorflowGraph is deprecated. "
"Will be removed in DeepChem 1.4.", DeprecationWarning)
self.graph = graph
self.session = session
self.name_scopes = name_scopes
self.output = output
self.labels = labels