tensorflow - Tensorflow 中的体积批量标准化

Question

Torch 在链接 中提供的体积批量标准化的 tensorflow 等效项是什么nn.VolumetricBatchNormalization？

我目前正在使用：

output = tf.contrib.layers.batch_norm(input, is_training=phase_train)

这是维度输入的正确使用方式(batch_size,height,width,depth,channels)吗？

Answer 1

Tensorflowtf.contrib.layers.batch_norm还支持体积批量归一化或 3D 批量归一化。

正确的输入维度是[batch, depth, height, width, in_channels]