2017-04-09 2 views
1

私はPython 2.7でTensorflowを使用しています。1階テンソルの要素を対応するインデックスで除算する

私はこのようなバイナリ1-Dテンソル有する:私はcorrispondentインデックスでテンソルの各元素の比率の合計を計算したい

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

を。

例えば、私は計算する必要があります。

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

はそれを行うTensorflow機能はありますか?

答えて

1

テンソルサイズ。

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  ]) 
1

# a is your 1-D binary tensor 
r = tf.range(1, a.get_shape()[0].value) 
result = a/r 

のようにあなたは、tf.rangeおよび/演算子を使用することができます。あなたが取得するtf.shapeまたはDzjkbの答えとしてget_shapeを使用することができます AおよびRは、同じタイプでなければなりません

関連する問題