a_shape = args[0].shape