blocks_test.py 文件源码

def test_forward_declaration_orphaned_nested(self):
    fwd1 = tdb.ForwardDeclaration(tdt.VoidType(), tdt.TensorType([]))
    fwd2 = tdb.ForwardDeclaration(tdt.SequenceType(tdt.TensorType([])),
                                  tdt.TensorType([]))
    b = tdb.Map(tdb.Scalar()) >> fwd2() >> tdb.Function(tf.negative)
    fwd2.resolve_to(tdb.Fold(tdb.Function(tf.add), fwd1()))
    fwd1.resolve_to(tdb.FromTensor(tf.ones([])))
    self.assertBuilds(-8., b, [3, 4], max_depth=3)