test_top_level4.py

import nnvm.symbol as sym

def test_binary_broadcast():
    x = sym.Variable('x')
    y = sym.Variable('y')
    z = x + y
    z = x * y
    z = x - y
    z = x / y


def test_broadcast_to():
    x = sym.Variable('x')
    y = sym.broadcast_to(x, shape=(3, 3))
    assert y.list_input_names() == ["x"]


if __name__ == "__main__":
    test_binary_broadcast()
    test_broadcast_to()