Loading deepchem/models/tensorgraph/layers.py +10 −0 Original line number Diff line number Diff line Loading @@ -645,8 +645,18 @@ class Reshape(Layer): class Cast(Layer): """ Wrapper around tf.cast. Changes the dtype of a single layer """ def __init__(self, in_layers=None, dtype=None, **kwargs): """ Parameters ---------- dtype: tf.DType the dtype to cast the in_layer to e.x. tf.int32 """ if dtype is None: raise ValueError("Must cast to a dtype") self.dtype = dtype Loading Loading
deepchem/models/tensorgraph/layers.py +10 −0 Original line number Diff line number Diff line Loading @@ -645,8 +645,18 @@ class Reshape(Layer): class Cast(Layer): """ Wrapper around tf.cast. Changes the dtype of a single layer """ def __init__(self, in_layers=None, dtype=None, **kwargs): """ Parameters ---------- dtype: tf.DType the dtype to cast the in_layer to e.x. tf.int32 """ if dtype is None: raise ValueError("Must cast to a dtype") self.dtype = dtype Loading
