code: def _setup(device, batch_size):
      t0, t1 = torch.tensor([0.0, 1.0], device=device)
      w0 = torch.zeros(batch_size, D, device=device)
      t = torch.rand([]).to(device)
      bm = torchsde.BrownianTree(t0=t0, t1=t1, w0=w0, entropy=0)
      return t, bm
  
file_path: torchsde/tests/test_brownian_tree.py
imports: import sys
  import numpy
  import numpy.random
  import torch
  from scipy.stats import norm, kstest
  import pytest
  import torchsde
label_desc: Use image node_id 1 for calling a global function with example usage: _setup(device, batch_size) and returns: t,  bm