torch.nn.functional.group_norm

torch.nn.functional.group_norm(input, num_groups, weight=None, bias=None, eps=1e-05) [исходный код]

Применяет групповую нормализацию к последнему заданному числу измерений.

Подробнее см. GroupNorm.

Тип возвращаемого значения:

Tensor