Commit 614d8a3c authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #766 from peastman/bug 

Bug fix to converting layers to tensors
parents 59a229a5 3162ce78

deepchem/models/tensorgraph/layers.py

+4 −1
Original line number Diff line number Diff line
@@ -162,7 +162,10 @@ class Layer(object): 
      tf.summary.histogram(self.name, self.tb_input, self.collections)



  def _as_graph_element(self):

    if '_as_graph_element' in dir(self.out_tensor):

      return self.out_tensor._as_graph_element()

    else:

      return self.out_tensor





def _convert_layer_to_tensor(value, dtype=None, name=None, as_ref=False):

CRA Git | Maintained and supported by SUSTech CRA and CCSE