我有一个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)
答案 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