class_docstr: split on dim=0 and store on multiple devices
class_docstr_tok: ["split","on","dim=0","and","store","on","multiple","devices"]
code: def split_device(self, other, dim=0):
      split_sections = [x.shape[0] for x in self.nphi]
      out = torch.split(other, split_sections, dim)
      return [y.to(x.device) for x, y in zip(self.nphi, out)]
  
imports: import collections
  import functools
  import io
  import os
  import warnings
  import dgl
  import numpy
  import scipy
  import torch
label_desc: Use image node_id 2 for calling the _Distributed obj's underlying member method code with example usage: obj.split_device(other, dim) and returns: unknown