Python mean_dimの例

コード例 #1

ファイル: act_norm.py プロジェクト: veroveroxie/alignflow

    def _get_moments(self, x):
        mean = mean_dim(x.clone(), dim=[0, 2, 3], keepdims=True)
        var = mean_dim((x.clone() - mean)**2, dim=[0, 2, 3], keepdims=True)
        log_std = torch.log(
            1. / (var.sqrt() + self.eps) / self.log_scale) * self.log_scale

        return mean, log_std

コード例 #2

ファイル: act_norm.py プロジェクト: turnip001/glow-1

    def initialize_parameters(self, x):
        if not self.training:
            return

        with torch.no_grad():
            bias = -mean_dim(x.clone(), dim=[0, 2, 3], keepdims=True)
            v = mean_dim((x.clone() + bias) ** 2, dim=[0, 2, 3], keepdims=True)
            logs = (self.scale / (v.sqrt() + self.eps)).log()
            self.bias.data.copy_(bias.data)
            self.logs.data.copy_(logs.data)
            self.is_initialized += 1.

コード例 #3

ファイル: act_norm.py プロジェクト: syakhmi/cs236-proj

    def _get_moments(self, x):
        mean = mean_dim(x.clone(), dim=[0, 2, 3], keepdims=True)
        var = mean_dim((x.clone() - mean) ** 2, dim=[0, 2, 3], keepdims=True)
        inv_std = 1. / (var.sqrt() + self.eps)

        return mean, inv_std