python - Theano：为什么在这种情况下索引会失败？

Question

我正在尝试获取给定布尔值的向量的最大值。

使用 Numpy：

>>> this = np.arange(10)
>>> this[~(this>=5)].max()
4

但是对于 Theano：

>>> that = T.arange(10, dtype='int32')
>>> that[~(that>=5)].max().eval()
9
>>> that[~(that>=5).nonzero()].max().eval()
Traceback (most recent call last):
  File "<pyshell#146>", line 1, in <module>
    that[~(that>=5).nonzero()].max().eval()
AttributeError: 'TensorVariable' object has no attribute 'nonzero'

为什么会这样？这是我缺少的细微差别吗？

Answer 1

您使用的是太旧的 Theano 版本。事实上， tensor_var.nonzero() 不在任何已发布的版本中。您需要更新到开发版本。

对于开发版本，我有这个：

>>> that[~(that>=5).nonzero()].max().eval()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: bad operand type for unary ~: 'tuple'

这是因为您的行中缺少括号。这是好线：

>>> that[(~(that>=5)).nonzero()].max().eval()
array(9, dtype=int32)

但是我们仍然有意想不到的结果！问题是 Theano 不支持 bool。在 int8 上做 ~ 是在 8 位上进行按位反转，而不是 1 位。它给出了这个结果：

>>> (that>=5).eval()
array([0, 0, 0, 0, 0, 1, 1, 1, 1, 1], dtype=int8)
>>> (~(that>=5)).eval()
array([-1, -1, -1, -1, -1, -2, -2, -2, -2, -2], dtype=int8)

您可以使用以下命令删除 ~：

>>> that[(that<5).nonzero()].max().eval()
array(4, dtype=int32)