GroupNorm

import torch
import torch.nn as nn

#使能KDNN
torch._C._set_kdnn_enabled(True)

# 输入数据: (batch_size, channels, height, width)
input = torch.randn(2, 6, 32, 32)  # 2张图片，6个通道，32x32分辨率 默认为fp32数据类型

# 定义 GroupNorm (分3组，每组2个通道)
group_norm = nn.GroupNorm(num_groups=3, num_channels=6)  # 6个通道分成3组 默认为fp32数据类型

# 前向计算
output = group_norm(input)

# 验证: 每组内归一化
print("GroupNorm 输出形状:", output.shape)  # 保持原形状