我理解tf.where将返回True值的位置,以便我可以使用结果shape[0]来获取Trues 的数量.
但是,当我尝试使用它时,维度是未知的(这是有意义的,因为它需要在运行时计算).所以我的问题是,我如何访问一个维度并在一个像总和的操作中使用它?
例如:
myOtherTensor = tf.constant([[True, True], [False, True]])
myTensor = tf.where(myOtherTensor)
myTensor.get_shape() #=> [None, 2]
sum = 0
sum += myTensor.get_shape().as_list()[0] # Well defined at runtime but considered None until then.
Run Code Online (Sandbox Code Playgroud)