def test_broadcast_dims():
    test((1, 2, 3))
    test((2, 1, 3))
    test((2, 3, 1))
    test2((1, 2, 3))
    test2((2, 1, 3))
    test2((2, 3, 1))
Test with some dimensions being 1.
dataset/ETHPy150Open rizar/attention-lvcsr/libs/Theano/theano/misc/tests/test_gnumpy_utils.py/test_broadcast_dims