code: def derivativeX(self,*args):
          '''
          Returns the derivative of the function with respect to the X dimension.
          This is the first input whenever n_dims < 4 and the second input otherwise.
          '''
          if self.n_dims >= 4:
              j = 1
          else:
              j = 0
          if self.i_dim == j:
              return np.ones_like(*args[0])
          else:
              return np.zeros_like(*args[0]f derivativeX(self,*args):
          '''
          Returns the derivative of the function with respect to the X dimension.
          This is the first input whenever n_dims < 4 and the second input otherwise.
          '''
          if self.n_dims >= 4:
              j = 1
          else:
              j = 0
          if self.i_dim == j:
              return np.ones_like(*args[0])
          else:
              return np.zeros_like(*args[0])
docs: Returns the derivative of the function with respect to the X dimension.
          This is the first input whenever n_dims < 4 and the second input otherwise.