* :obj:`~nnabla.Variable`: Std (if ``output_stat=True`)
shape_stat = [1 for _ in range(len(inp) + 1)]
shape_stat[batch_axis] = inp.shape[batch_axis]
shape_stat[channel_axis] = num_groups
shape_stat[channel_axis + 1] = int(inp.shape[channel_axis] / num_groups)
if param_init is None:
After Change
from nnabla.normalization import _force_list
batch_axis = _force_list(batch_axis)
shape_stat = [inp.shape[i]
if i in batch_axis else 1 for i in range(len(inp.shape) + 1)]
shape_stat[channel_axis] = num_groups
shape_stat[channel_axis + 1] = int(inp.shape[channel_axis] / num_groups)
