def test_all_of_different_shapes(self):
block = scalar_all_of(tf.negative, functools.partial(tf.expand_dims, dim=1))
self.assertBuilds((-3., [3.]), block, 3)
微信
公众号
扫码关注公众号