官方API文档中只提供了2维和4、5维应用批归一化算子，请问我在2维基础上还有额外1维batch构成的3维数据没有直接的算子吗，只能按batch维对那两维一个个使用BatchNorm1d()吗