python - 如何在 Theano 中进行 0/1 子集化？

Question

目标是通过另一个数组中提供的值获取元素数组的子集。

import theano
import theano.tensor as T

a = T.vector('X', dtype='int64')
b = T.vector('Y', dtype='int64')
c = a[b]
g = function([a,b],c)

x = np.array([5,3,2,3,4,6], dtype=int)
y = np.array([0,0,1,0,0,1], dtype=int)
print g(x,y)

这会打印

[5 5 3 5 5 3]

而不是

[2 6]

如何获得预期的结果？

Answer 1

尝试使用nonzero()函数。

您的案例示例:

import theano
import theano.tensor as T

a = T.vector('X', dtype='int64')
b = T.vector('Y', dtype='int64')
c = a[b.nonzero()]
g = function([a,b],c)

x = np.array([5,3,2,3,4,6], dtype=int)
y = np.array([0,0,1,0,0,1], dtype=int)
print g(x,y)

希望对你有帮助