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

Merge pull request #569 from peastman/axis 

Fixed bug in ReduceMean layer
parents 2612ceb8 d03f1e22

deepchem/models/tensorgraph/layers.py

+1 −1
Original line number Diff line number Diff line
@@ -640,7 +640,7 @@ class ReduceMean(Layer): 
    else:

      self.out_tensor = inputs[0]



    out_tensor = tf.reduce_mean(self.out_tensor)

    out_tensor = tf.reduce_mean(self.out_tensor, axis=self.axis)

    if set_tensors:

      self.out_tensor = out_tensor

    return out_tensor

CRA Git | Maintained and supported by SUSTech CRA and CCSE