code: def get_cells(start, stop, nbr_cutoff, ndim=3):
      """Returns the locations of all grid points in box.
  
      Suppose start is -10 Angstrom, stop is 10 Angstrom, nbr_cutoff is 1.
      Then would return a list of length 20^3 whose entries would be
      [(-10, -10, -10), (-10, -10, -9), ..., (9, 9, 9)]
  
      Returns
      -------
      cells: tf.Tensor
        (box_size**ndim, ndim) shape.
      """
      ranges = [tf.range(start, stop, nbr_cutoff) for _ in range(ndim)]
      return tf.reshape(
          tf.transpose(tf.stack(tf.meshgrid(*ranges))), (-1, ndim)
      )
  
imports: from __future__ import print_function
  from __future__ import division
  from __future__ import unicode_literals
  import warnings
  import numpy
  import tensorflow
  from deepchem.models import Model
  from deepchem.nn import model_ops
  import deepchem.utils.rdkit_util
label_desc: Use image node_id 4 for calling a global function with example usage: get_cells(start, stop, nbr_cutoff, ndim) and returns: tf