转换后无法更新共享张量变量的子集

Question

我有以下代码：

import theano.tensor as T

Words = theano.shared(value = U, name = 'Words')
zero_vec_tensor = T.vector()
zero_vec = np.zeros(img_w, dtype = theano.config.floatX)
set_zero = theano.function([zero_vec_tensor], updates=[(Words, T.set_subtensor(Words[0,:], zero_vec_tensor))])

编译正常（其中 U 是 dtype float64 的 numpy 数组）。

为了防止将来出现类型错误，我想将共享张量 Words 转换为 float32（或 theano.config.floatX，这与我将 floatX 设置为 float32 在配置文件中）。

我添加 Words = T.cast(Words, dtype = theano.config.floatX) 然后我得到以下错误：

TypeError: ('update target must be a SharedVariable', Elemwise{Cast{float32}}.0)。

我不明白为什么。根据此 question，使用 set_subtensor 应该允许我更新共享变量的一个子集。

如何在能够更新共享张量的同时投射它？

Answer 1

问题是您正在尝试更新符号变量，而不是共享变量。

U = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.float64)
Words = theano.shared(value=U, name='Words')
zero_vec_tensor = T.vector()
set_zero = theano.function([zero_vec_tensor], updates=[(Words, T.set_subtensor(Words[0, :], zero_vec_tensor))])

工作正常，因为您正在更新的东西 Words 是一个共享变量。

U = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.float64)
Words = theano.shared(value=U, name='Words')
Words = T.cast(Words, dtype = theano.config.floatX)
zero_vec_tensor = T.vector()
set_zero = theano.function([zero_vec_tensor], updates=[(Words, T.set_subtensor(Words[0, :], zero_vec_tensor))])

不起作用，因为现在 Words 不再是共享变量，它是一个符号变量，在执行时，将计算将共享变量中的值转换为 theano.config.floatX。

共享变量的dtype由分配给它的值决定。所以你可能只需要改变 U:

的类型

U = np.array([[1, 2, 3], [4, 5, 6]], dtype=theano.config.floatX)

或者使用 numpy 而不是符号方式进行转换：

U = np.dtype(theano.config.floatX).type(U)

转换后无法更新共享张量变量的子集

Can not update a subset of shared tensor variable after a cast

numpy

python-2.7

theano