Как обрабатывать тип anano

Question

У меня есть матрица 'y', и я хочу добавить одно из чисел в каждую строку y. Затем я устанавливаю матрицу 'x', в которой первый столбец устанавливает тот, который я хочу Добавить. Я использую следующий код:

import numpy 
import theano 
import theano.tensor as T 

x = T.matrix('x') 
y = T.matrix('y') 
z = T.sum(y[T.arange(1,5),x[:,0]]) 

f = theano.function(inputs=[x,y],outputs=z) 

x = numpy.asarray([[1],[2],[3],[4],[0]]) 
y = numpy.ones(5,5) 
print(f([x,y])) 

Но это cann't работы. Будет ошибка грамматики. почему не работает этот код?

Answer 1

есть некоторые проблемы в вашем коде. первых, определить свой тип данных тщательно, а затем Becareful при использовании [] кронштейна (посмотреть, как вы просто сделать еще один массив при вызове функции с помощью f([x,y]))

так, после того, как я изменить код стал:

import numpy 
import theano 
import theano.tensor as T 

x = T.imatrix('x') #define integer matrix 
y = T.imatrix('y') 
z = T.sum(y[T.arange(5),x[:,0]]) 

f = theano.function(inputs=[x,y],outputs=z) 

x = numpy.asarray([[1],[2],[3],[4],[0]]) 
y = numpy.ones((5,5),dtype="int32") #define integer array 
print(f(x,y)) #remove square bracket 

Надеюсь, это поможет