Ejemplo n.º 1
0
 def __init__(self, input_channel, out_channel):
     super(BroadCastNet, self).__init__()
     self.dense = Dense(input_channel, out_channel)
     self.broadcast = Broadcast(0)
Ejemplo n.º 2
0
def test_broadcast(x):
    print('test_broadcast with %s' % (x))
    broadcast = Broadcast(0)
    y = broadcast(x)
    print('y=%s' % (y))