pytorch BatchNorm1d 输入二维和三维数据的区别
在阅读KPConv-PyTorch源码时,发现其对torch.nn.BatchNorm1d进行了封装。class BatchNormBlock(nn.Module): def __init__(self, in_dim, use_bn, bn_momentum): """ Initialize a batch normalization block. If network does not use batch normalizati