在tensorflow中访问python列表和张量的一种优雅方法

时间:2019-02-03 01:09:35

标签: python tensorflow

Python列表,例如a = [1, 2, 3, 4]可以访问a[1]。但是我想访问除index = 1以外的值,也就是说,我想访问a[0]a[3:]。有没有一种优雅的方法来访问列表(例如a[~1])可以解决此问题?

实际上,我想在tensorflow中访问张量。我知道一些我不想访问的索引,而我需要所有其他索引。尽管我们可以通过一些乏味的方法来解决此问题,例如将变量链接重新定义为其他索引,但这并不是我们真正想要的。

1 个答案:

答案 0 :(得分:1)

这里:

# The index that you want to exclude
mask = tf.not_equal(tf.range(tf.shape(X)[0]), index)
masked = tf.boolean_mask(X, mask)