0

我有以下代码

from tensorflow import keras
from keras import backend as K

pool_size = (2,2,2)
strides = (2,2,2)

yt = K.zeros(shape=(10,10,10))

result = keras.backend.pool3d(yt, pool_size, strides, pool_mode="avg")

当我尝试运行它说的代码时

.. InvalidArgumentError: tensor_in 必须是 5 维 [Op:AvgPool3D] 名称：AvgPool3D/

我似乎不喜欢 yt 的维度。但我想在尺寸为 3x3x3 的 3d 图像中最大化池。另一个维度应该是什么？

4

1 回答 1

1
  • 批量大小
  • 渠道

正如 Keras 中的每个类似卷积的操作一样，这些维度都是必需的。

  • 使用“channels_last”（默认）：(batch, size1, size2, size3, channels)
  • 使用“channels_first”：(batch, channels, size1, size2, size3)

所以：

yt = K.zeros(shape=(1,10,10,10,1))
于 2019-12-06T18:18:56.390 回答