代码片段，生成高斯噪声

最新推荐文章于 2025-03-07 19:58:47 发布

图像生成小菜鸟

最新推荐文章于 2025-03-07 19:58:47 发布

阅读量157

点赞数

文章标签： python

本文链接：https://blog.youkuaiyun.com/qq_28234239/article/details/135088972

版权

    def get_z0(self, batch, train=True):
      n,c,h,w = batch.shape 

      if self.init_type == 'gaussian':
          ### standard gaussian #+ 0.5
          cur_shape = (n, c, h, w)
          return torch.randn(cur_shape)*self.noise_scale
      else:
          raise NotImplementedError("INITIALIZATION TYPE NOT IMPLEMENTED")