如何在Tensorflow中使用3d卷积？

import tensorflow as tf

input_data = tf.placeholder(tf.float32, shape=[batch_size, depth, height, width, channels])
filter_weights = tf.Variable(tf.truncated_normal([filter_depth, filter_height, filter_width, in_channels, out_channels], stddev=0.1))

conv3d = tf.nn.conv3d(input_data, filter_weights, strides=[1, stride_depth, stride_height, stride_width, 1], padding='SAME')

bias = tf.Variable(tf.constant(0.1, shape=[out_channels]))
conv3d_with_bias = tf.nn.bias_add(conv3d, bias)
output = tf.nn.relu(conv3d_with_bias)