def random(self, *shapes, **kwargs):
if all(isinstance(i, int) for i in shapes):
if kwargs.get("complex", False):
return (self.random(*shapes) + 1j * self.random(*shapes)).astype(np.complex64)
else:
return np.random.rand(*shapes)
else:
return tuple(self.random(*shape) for shape in shapes)
评论列表
文章目录