官方API文档中只提供了2维和4、5维应用批归一化算子,请问我在2维基础上还有额外1维batch构成的3维数据没有直接的算子吗,只能按batch维对那两维一个个使用BatchNorm1d()吗
****************************************************解答*****************************************************
对于3d的数据(N, C, L),如果要在L维度做batch-norm,可以将前两个维度进行合并,reshape到(N*C, L),调用1d,之后对结果进行Reshape。如果要对C,L操作,可以reshape到(N, C*L)然后调用1D。之后对结果再进行reshape。
对于输入size和输出size可以整数的情况,可以先计算kernel_size再使用AvgPool进行等价计算。