pool_2d в Theano выдает ошибку несоответствия размеров

Question

Испытывая метод 'pool_2d' в Theano, я получил ошибку "Неверное количество измерений: ожидается 4, получено 1 с формой (16,)". Код приведен ниже.

from theano import tensor, shared, function
from theano.tensor.signal.pool import pool_2d
import numpy


class Test:
    def __init__(self):
        i = tensor.dtensor4()
        o = pool_2d(input = i, ws = (2, 2), ignore_border = True, stride = None, pad = (0, 0), mode = 'max')

        self.pool = function([i], o)

    def pool(self, iput):
        iput = numpy.array(iput)
        iput.shape = (1, 1, 4, 4)
        print(self.pool(iput))


x = Test()
x.pool([1.,2.,3.,4.,5.,6.,7.,8.,9.,10.,11.,12.,13.,14.,15.,16.])

В чем здесь причина ошибки?

Answer 1

Это была глупая ошибка, которую я сделал. Я назначил функцию Theano переменной self.pool, а затем создал другой метод с тем же именем, что и у 'pool', вызывая имя cla sh.