python - Theano 函数使用 'givens' 属性引发 ValueError

Question

我使用 theano 函数并希望使用 givens 迭代所有输入样本。代码如下:

index = T.scalar('index')
train_set = np.array([[0.2, 0.5, 0.01], [0.3, 0.91, 0.4], [0.1, 0.7, 0.22], 
                      [0.7, 0.54, 0.2], [0.1, 0.12, 0.3], [0.2, 0.52, 0.1], 
                      [0.12, 0.08, 0.4], [0.02, 0.7, 0.22], [0.71, 0.5, 0.2], 
                      [0.1, 0.42, 0.63]])
train = function(inputs=[index], outputs=cost, updates=updates, 
                 givens={x: train_set[index]})

它最终引发了一个错误:

ValueError: setting an array element with a sequence.

你能告诉我为什么，以及如何解决这个问题吗？

Answer 1

问题是这样的:train_set[index]

这里的 train_set 是一个 numpy ndarray，索引是一个 Theano 变量。 NumPy 不知道如何使用 Theano 变量。您必须将 train_set 转换为 Theano 变量，如共享变量:

train_set = theano.shared(train_set)

您还需要更改索引声明，因为 Theano 不支持索引的实际值:

index = T.iscalar()