class_docstr: A wrapper around torchsde.BrownianTree that enables batches of entropy.
class_docstr_tok: ["A","wrapper","around","torchsde.BrownianTree","that","enables","batches","of","entropy","."]
code: def sort(a, b):
      return (a, b, 1) if a < b else (b, a, -1)
  
imports: import math
  import torch
  import torchsde
  from scipy import integrate
  from torch import nn
  from torchdiffeq import odeint
  from tqdm.auto import tqdm, trange
label_desc: Use image node_id 2 for calling the BatchedBrownianTree obj's underlying member method code with example usage: obj.sort(a, b) and returns: unknown