本文整理匯總了Python中tensorflow_fold.ForwardDeclaration方法的典型用法代碼示例。如果您正苦於以下問題：Python tensorflow_fold.ForwardDeclaration方法的具體用法？Python tensorflow_fold.ForwardDeclaration怎麽用？Python tensorflow_fold.ForwardDeclaration使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類tensorflow_fold的用法示例。
在下文中一共展示了tensorflow_fold.ForwardDeclaration方法的2個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: dynamic_pooling_blk
# 需要導入模塊: import tensorflow_fold [as 別名]
# 或者: from tensorflow_fold import ForwardDeclaration [as 別名]
def dynamic_pooling_blk():
"""Input: root node dic
Output: pooled, TensorType([hyper.conv_dim, ])
"""
leaf_case = feature_detector_blk()
pool_fwd = td.ForwardDeclaration(td.PyObjectType(), td.TensorType([hyper.conv_dim, ]))
pool = td.Composition()
with pool.scope():
cur_fea = feature_detector_blk().reads(pool.input)
children = td.GetItem('children').reads(pool.input)
mapped = td.Map(pool_fwd()).reads(children)
summed = td.Reduce(td.Function(tf.maximum)).reads(mapped)
summed = td.Function(tf.maximum).reads(summed, cur_fea)
pool.output.reads(summed)
pool = td.OneOf(lambda x: x['clen'] == 0,
{True: leaf_case, False: pool})
pool_fwd.resolve_to(pool)
return pool 示例2: tree_sum_blk
# 需要導入模塊: import tensorflow_fold [as 別名]
# 或者: from tensorflow_fold import ForwardDeclaration [as 別名]
def tree_sum_blk(loss_blk):
# traverse the tree to sum up the loss
tree_sum_fwd = td.ForwardDeclaration(td.PyObjectType(), td.TensorType([]))
tree_sum = td.Composition()
with tree_sum.scope():
myloss = loss_blk().reads(tree_sum.input)
children = td.GetItem('children').reads(tree_sum.input)
mapped = td.Map(tree_sum_fwd()).reads(children)
summed = td.Reduce(td.Function(tf.add)).reads(mapped)
summed = td.Function(tf.add).reads(summed, myloss)
tree_sum.output.reads(summed)
tree_sum_fwd.resolve_to(tree_sum)
return tree_sum