TensorFlow函数:tf.nn.avg_pool3d

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.

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部