Consider to add emulation to support multiple axes case, e.g. input shape is [2, 3, 4, 5] and scale shape is [1, 3, 4, 1].