Tensorflow（Regression）：如何在损失函数中实现一个函数

Question

我想设计一个NN，其中NN的输出插入到我的自定义函数中（例如，根据NN的输出计算某些东西），而在损失函数中，函数的测量和输出之间的误差被最小化。

在我的脚本中

y_pred = tf.contrib.layers.fully_connected(h4, 4, tf.nn.relu) 
#y_pred should be four numbers

loss = tf.reduce_mean(tf.squared_difference(y,myfunction(y_pred[:,0],y_pred[:,1],y_pred[:,2],y_pred[:,3])))

那我有

ValueError: setting an array element with a sequence

您会为我的问题推荐任何解决方案吗？