Commit d03f1e22 authored by peastman's avatar peastman
Browse files

Fixed bug in ReduceMean layer

parent 2612ceb8

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