'Get the index-th output.
Parameters
index : int
The index of output.
Returns
out : numpy array.
The output array.'

def get_output(self, index):

    pdata = ctypes.POINTER(mx_uint)()
    ndim = mx_uint()
    _check_call(_LIB.MXPredGetOutputShape(self.handle, index, ctypes.byref(pdata), ctypes.byref(ndim)))
    shape = tuple(pdata[:ndim.value])
    data = np.empty(shape, dtype=np.float32)
    _check_call(_LIB.MXPredGetOutput(self.handle, mx_uint(index), data.ctypes.data_as(mx_float_p), mx_uint(data.size)))
    return data
