class_docstr: The multi dimensional Brownian Motion.
class_docstr_tok: ["The","multi","dimensional","Brownian","Motion","."]
docstr: The integrated drift of the process.
  
  Returns:
    None or a Python callable. None is returned if the input drift was a
    callable and no total drift function was supplied.
    The callable returns the integrated drift rate between two times.
    It accepts two real `Tensor` arguments. The first argument is the
    left end point and the second is the right end point of the time interval
    for which the total drift is needed. Both the `Tensor` arguments are of
    the same dtype and shape. The return value of the callable is
    a real `Tensor` of the same dtype as the input arguments and of shape
    `times_shape + [dim]` where `times_shape` is the shape of the times
    `Tensor`.
docstr_tok: ["The","integrated","drift","of","the","process",".","Returns",":","None","or","a","Python","callable",".","None","is","returned","if","the","input","drift","was","a","callable","and","no","total","drift","function","was","supplied",".","The","callable","returns","the","integrated","drift","rate","between","two","times",".","It","accepts","two","real","`","Tensor","`","arguments",".","The","first","argument","is","the","left","end","point","and","the","second","is","the","right","end","point","of","the","time","interval","for","which","the","total","drift","is","needed",".","Both","the","`","Tensor","`","arguments","are","of","the","same","dtype","and","shape",".","The","return","value","of","the","callable","is","a","real","`","Tensor","`","of","the","same","dtype","as","the","input","arguments","and","of","shape","`","times_shape","+","[","dim","]","`","where","`","times_shape","`","is","the","shape","of","the","times","`","Tensor","`","."]
code: def total_drift_fn(self):
      """The integrated drift of the process.
  
      Returns:
        None or a Python callable. None is returned if the input drift was a
        callable and no total drift function was supplied.
        The callable returns the integrated drift rate between two times.
        It accepts two real `Tensor` arguments. The first argument is the
        left end point and the second is the right end point of the time interval
        for which the total drift is needed. Both the `Tensor` arguments are of
        the same dtype and shape. The return value of the callable is
        a real `Tensor` of the same dtype as the input arguments and of shape
        `times_shape + [dim]` where `times_shape` is the shape of the times
        `Tensor`.
      """
      return self._total_drift_fn
  
file_path: tf-quant-finance/tf_quant_finance/models/legacy/brownian_motion.py
imports: import tensorflow.compat.v2
  from tf_quant_finance.math.random_ops import multivariate_normal
  from tf_quant_finance.models.legacy import brownian_motion_utils
  from tf_quant_finance.models.legacy import ito_process
label_desc: Use image node_id 7 for calling the BrownianMotion obj's underlying member method code with example usage: obj.total_drift_fn() and returns: self