用相应的指数

Question

我使用Tensorflow和python 2.7。

我有这样的二进制 1-D张量：

a = [ 1 , 0, 1, 0, 1]

我希望通过相应指数计算张量的每个元素的比率之和。

例如，我需要计算：

[ 1/1 + 0/2 + 1/3 + 0/4 + 0/5 ]

是否有 Tensorflow 功能吗？

Answer 1

您可以使用tf.range和/运算符，例如

group by

注意： a和r必须属于同一类型

Answer 2

您可以使用tf.shape或Dzjkb的答案get_shape来获取张量大小。

import tensorflow as tf
sess = tf.Session()
x = tf.constant([1,0,1,0,1])
y = tf.range(1, tf.shape(x)[0] + 1)
#y = tf.range(1, x.get_shape()[0].value + 1)
z = x / y
sess.run(z)

输出：

array([ 1.        ,  0.        ,  0.33333333,  0.        ,  0.2       ])