TensorFlow函数:tf.nn.avg_pool3d
2019-01-31 11:29 更新
tf.nn.avg_pool3d函数
tf.nn.avg_pool3d(
input,
ksize,
strides,
padding,
data_format='NDHWC',
name=None
)
定义在:tensorflow/python/ops/gen_nn_ops.py.
请参阅指南:神经网络>池操作
在输入上执行3D平均池.
参数:
- input:一个Tensor,并且必须是下列类型之一:half,bfloat16,float32,float64.形状[batch, depth, rows, cols, channels]张量到池.
- ksize:ints列表中有长度>= 5,长度为5的1-D张量,输入张量的每个维度的窗口大小,必须有:ksize[0] = ksize[4] = 1.
- strides:ints列表中有长度>= 5,长度为5的1-D张量,input中每个维度的滑动窗口的步幅.必须有:strides[0] = strides[4] = 1.
- padding:来自:"SAME", "VALID"的一个string.要使用的填充算法的类型.
- data_format:来自:"NDHWC", "NCDHW"的可选string,默认为"NDHWC",输入和输出数据的数据格式,使用默认格式“NDHWC”,数据按以下顺序存储:[batch,in_depth,in_height,in_width,in_channels];或者,格式可以是“NCDHW”,数据存储顺序是:[batch,in_channels,in_depth,in_height,in_width].
- name:操作的名称(可选).
返回:
与input具有相同的类型的Tensor.