pytorch：对张量进行批处理？

Question

我有一批图像，大小为(B,C,H,W)（手电筒。张量） 我想使用大小为(C,H,W)

的均值/标准张量（例如：[0.4,0.3,0.2]）对其进行归一化

我的问题是，Pytorch中是否有任何操作可以在Tensorflow中实现dataset.map或tf.expand_dims这样的功能？

注意，它不是PIL图像格式