blocks_test.py 文件源码

def test_composition_void(self):
    c = tdb.Composition()
    with c.scope():
      a = tdb.Scalar().reads(c.input)
      b = tdb.Function(tf.negative).reads(a)
      tdm.Metric('foo').reads(b)
      c.output.reads(a)
    self.assertBuilds((42., {'foo': [-42.]}), c, 42, max_depth=2)