(TypeError) точечная операция в numpy и theano

Question

Я пишу следующий код:

import theano
import numpy

p=theano.tensor.dmatrix('p')
q=theano.tensor.dmatrix('q')
r=theano.tensor.dot(p,q)

f=theano.function([p,q], r)

a=numpy.array([1,2])
b=numpy.array([[1,2,3],[4,5,6]])

затем numpy.dot(a,b) возвращается array([ 9, 12, 15]), Я думаю f(a,b) возвращает то же самое, но это не удается (TypeError). Сообщение об ошибке говорит

Неверное количество измерений: ожидается 2, получено 1 с формой (2,).

Что происходит?

Answer 1

Измените строку 4 на p=theano.tensor.dvector('p'), Theano строже печатает, чем тупит.