code: def _setup(device):
      t0, t1 = torch.tensor([0.0, 1.0], device=device)
      w0, w1 = torch.randn([2, BATCH_SIZE, D], device=device)
      t = torch.rand([], device=device)
      bm = torchsde.BrownianPath(t0=t0, w0=w0)
      return t, bm
  
file_path: torchsde/tests/test_brownian_path.py
imports: import sys
  import math
  import numpy
  import numpy.random
  import torch
  from scipy.stats import norm, kstest
  import torchsde
  import pytest
label_desc: Use image node_id 1 for calling a global function with example usage: _setup(device) and returns: t,  bm