def _polynomial(tensor):
size = int(tensor.get_shape()[1])
pows = [ tf.pow(tensor[:, n], n + 1) for n in range(size) ]
return tf.transpose(tf.pack(pows))
微信
公众号
扫码关注公众号