TensorFlow：在圖中使用張量作爲列表參數

我想使用tf.Tensor作爲TensorFlow圖中另一個操作的類型（python）列表的參數。換句話說，我想用一個張量作爲另一個操作的動態列表參數。這可能嗎？TensorFlow：在圖中使用張量作爲列表參數

可執行例如：

import tensorflow as tf 
import numpy as np 

graph = tf.Graph() 
var1 = np.random.randn(2, 3) 
var2 = np.random.randn(2, 3, 4) 

with graph.as_default(): 
    def getRange(myTensor): 
     myRank = tf.rank(myTensor) 
     return tf.range(tf.constant(1), tf.squeeze(myRank)) 

    def getMoments(myTensor): 
     myMoments = tf.nn.moments(myTensor, axes=getRange(myTensor)) 
     return myMoments 
    var1tf = tf.Variable(var1) 
    var2tf = tf.Variable(var2) 
    var1moments = getMoments(var1tf) 
    var2moments = getMoments(var2tf) 
    rangeVar1 = getRange(var1tf) 
    rangeVar2 = getRange(var2tf) 
    init = tf.global_variables_initializer() 

with tf.Session(graph=graph) as sess: 
    sess.run([init]) 
    print(sess.run([rangeVar1])) # outputs [1], ok 
    print(sess.run([rangeVar2])) # outputs [1, 2], ok 
    print(sess.run([var1moments])) 
    print(sess.run([var2moments]))

這拋出：

raise TypeError("'Tensor' object is not iterable.")

來源

2017-05-18 gizzmole

有趣的問題。只是爲了好奇，你有什麼樣的問題需要處理未知級別的張力？ – user1735003

我在用不同等級的張量調用的函數中使用此代碼 – gizzmole

將我的示例更新爲可執行示例 – gizzmole

我發現使用get_shape（）中的溶液：

def getMoments(myTensor): 
    myRank = len(myTensor.get_shape().as_list()) 
    print('rank via shape:', myRank) 
    myMoments = tf.nn.moments(myTensor, axes=list(range(1, myRank))) 
    return myMoments

來源

2017-06-06 06:00:53 gizzmole

TensorFlow：在圖中使用張量作爲列表參數

回答

相關問題