在tensorflow中聚合张量的每个元素

时间:2016-02-26 16:33:48

标签: tensorflow

我有一个3-D张量如下:

A=array([[[False, False, False],
        [False, False,  True],
        [False,  True,  True],
        [ True,  True,  True],
        [ True,  True, False]]], dtype=bool)
        A.shape= (1,5,3)

我想将它转换为图中张量流中的二维张量,其中if 我在任何元素中都有一个True,聚合结果应为True,否则就是False。如下:

output=array([[False, True, True, True, True]], dtype=bool)
output.shape= (1,5)

1 个答案:

答案 0 :(得分:1)

这应该完全符合您的要求(直接来自文档):

# 'x' is [[True,  True]
#         [False, False]]
tf.reduce_any(x) ==> True
tf.reduce_any(x, 0) ==> [True, True]
tf.reduce_any(x, 1) ==> [True, False]   <== this is what you need

https://www.tensorflow.org/versions/r0.7/api_docs/python/math_ops.html#reduce_any