[英]TensorFlow: use tensor as list argument in graph
我想將tf.Tensor用作TensorFlow圖中另一個操作的(python)列表類型的參數。 換句話說,我想將Tensor用作其他操作的動態列表參數。 這可能嗎?
可執行示例:
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.")
我找到了使用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
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.