给用户增加一个接口，用来指定batch在哪个维度

[whisper233]

比较常用的spikingjelly库的默认输入维度为[T,batchsize, c,h,w]，这与当前库的默认输入维度不同，当前库在计算batchsize时默认batch在0号维度上，这可能导致batch计算错误，这次修改增加了一个参数，用来指定batch在哪个维度上，方便后续的代码batch_counter的计数。代码加的不好哈，仅当参考

[NneurotransmitterR]

当使用dataloader时形状如果是常用的[N, C, H, W]，或像README只指定形状[C, H, W]，这两种情况下生成数据的形状都是[N, C, H, W]，好像并没有涉及到T维度，而且计算结果也只是一个时间步的？可能本仓库对多步模式没有很好的支持（包括README的示例中使用的spiking resnet使用的是单步输入），如果直接使用多步模式的输入[T, N, C, H, W]的话是会有这个问题。总之不同版本的SJ，推理代码中不同步进模式的使用，模型forward的实现以及对SNN的计算量的不同理解都可能导致对时间维度处理的不同，如果要与新版本的SpikingJelly上的最佳实践相兼容可能还是需要整体更新一下代码。