使用具有tensorflow对象的元组创建类对象

Posted 2021-03-28

我有一个参数Theta类，它创建神经网络如下：

class parametersTheta:
    def __init__(self, weight1, weight2,....):
        self.weightName1 = weight1
        self.weightName2 = weight2
        ...
        self.sess = tf.Session()
    def makeWorkerTheta(self, param):
        return parametersTheta(self.sess.run(functionCalculatingTensorWeights, feed_dict={...}))

self.sess.run创建了所有权重张量的元组。但是，会弹出错误提示您需要输入weight2及以后，即元组进入weight1

我怎么解决这个问题？基本上，我如何用元组创建类参数Theta的实例？

答案

您可以使用扩展为这样的参数的元组实例化类。

parametersTheta(*(weight1, weight2, ...))

元组之前的星号将其扩展为相应的参数列表。