Skip to main content
 首页 » 编程设计

python之在 tensorflow 中访问 python 列表和张量的优雅方式

2024年12月31日56tuyile006

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

其实我想访问tensorflow中的tensor。我知道几个我不想访问的索引,我需要所有其他索引。虽然我们可以通过一些繁琐的方法来解决这个问题,比如重新定义一个指向其他索引的变量链接,但这并不是我们真正想要的。

请您参考如下方法:

这里:

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