Example #1
0
    def testBroadcastStructure(self):
        if not tf.executing_eagerly():
            return
        struct = fun_mcmc.maybe_broadcast_structure(1, [1, 2])
        self.assertEqual([1, 1], struct)

        struct = fun_mcmc.maybe_broadcast_structure([3, 4], [1, 2])
        self.assertEqual([3, 4], struct)
Example #2
0
    def testBroadcastStructure(self):
        struct = fun_mcmc.maybe_broadcast_structure(1, [1, 2])
        self.assertEqual([1, 1], struct)

        struct = fun_mcmc.maybe_broadcast_structure([3, 4], [1, 2])
        self.assertEqual([3, 4], struct)