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 __init__(self, prior, nphi):
      self.prior = prior
      self.nphi = nphi
      super().__init__(
          [_Dirichlet(self.prior, nphi) for nphi in self.nphi]
      )
  
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 1 to create a new _Distributed object from inherited base classes: collections with example: obj = _Distributed(prior, nphi)