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] name: AvgPool3D/

yt のディメンションが気に入らないようです。しかし、寸法が 3x3x3 の 3D イメージでプールを最大化したい。他の次元はどうあるべきですか?

4

1 に答える 1