64099a1d2d671705cd9d4a18f9413f4c3569bc86,deepchem/models/tensorgraph/layers.py,NeighborList,get_cells,#NeighborList#,3641

Before Change

    cells: tf.Tensor
      (n_cells, ndim) shape.
    
    start, stop, nbr_cutoff = self.start, self.stop, self.nbr_cutoff
    mesh_args = [tf.range(start, stop, nbr_cutoff) for _ in range(self.ndim)]
    return tf.cast(
        tf.reshape(
            tf.transpose(tf.stack(tf.meshgrid(*mesh_args))),
            (self.n_cells, self.ndim)), tf.float32)


class Dropout(Layer):

After Change

    cells: tf.Tensor
      (n_cells, ndim) shape.
    
    return self._get_layer(False).get_cells()


class Dropout(Layer):

Instances