Tensorflow中的體積批量標準化

Question

Torch在linknn.VolumetricBatchNormalization中提供的體積批量標準化的張量等效量是多少？

我目前正在使用：

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

這是使用了尺寸(batch_size,height,width,depth,channels)輸入的正確方法是什麼？

Answer 1

Tensorflow tf.contrib.layers.batch_norm也支持批量批量標準化或3D批量標準化。

正確的輸入尺寸是[batch, depth, height, width, in_channels]