2e7d63a19355f8d9e414b1d1eef24475a7b63a14,python/test/function/test_instance_normalization.py,,ref_instance_normalization,#Any#Any#Any#Any#Any#Any#Any#,7

Before Change

def ref_instance_normalization(x, beta, gamma, channel_axis, batch_axis, eps, output_stat):

    if hasattr(batch_axis, "__iter__"):
        ignore_axes = batch_axis + [channel_axis, ]
    else:
        ignore_axes = [batch_axis, channel_axis]

    axes = tuple([i for i in range(len(x.shape)) if i not in ignore_axes])

    x_mean = x.mean(axis=axes, keepdims=True)
    x_std = x.std(axis=axes, keepdims=True)

After Change

def ref_instance_normalization(x, beta, gamma, channel_axis, batch_axis, eps, output_stat):

    ignore_axes = _force_list(batch_axis) + [channel_axis, ]

    axes = tuple(_get_axes_excluding(len(x.shape), ignore_axes))

Instances