如何获得包含张量中每个元素的列表？

Question

我正在学习tf.contrib.opt.ScipyOptimizerInterface，并在演示中限制每个权重都是非负的。

# -*- coding: utf-8 -*-
import tensorflow as tf

vector = tf.constant([.021,.046,.013], name='vector')
wt = tf.Variable([1./3,1./3,1./3], 'wt')

loss = -tf.reduce_sum(tf.multiply(vector,wt,'loss'))
equalities = [tf.reduce_sum(wt) - 1.]
inequalities = [wt[0],wt[1],wt[2]]

optimizer = tf.contrib.opt.ScipyOptimizerInterface(loss, var_list=[wt], equalities=equalities, inequalities=inequalities, method='SLSQP')

with tf.Session() as session:
  session.run(tf.global_variables_initializer())
  optimizer.minimize(session)

ScipyOptimizerInterface

equalities：要保持等于零的等式约束标量张量的可选列表。inequalities：保持非负的不等式约束标量张量的可选列表。

如何将inequalities = [wt[0],wt[1],wt[2]]更改为类似于inequalities = [wt[i] for i in range(tf.size(weight))]的内容？

Answer 1

您可以将其设置为：

inequalities = [wt[i] for i in range(wt.get_shape()[0])]