Source code for pycomposer.samplabledata.truesampler._mxnet.midi_true_sampler

# -*- coding: utf-8 -*-
import numpy as np
from pycomposer.samplabledata.truesampler.midi_true_sampler import MidiTrueSampler as _MidiTrueSampler
import mxnet as mx
import mxnet.ndarray as nd


class MidiTrueSampler(_MidiTrueSampler):
    '''
    Sampler which draws samples from the `true` distribution of MIDI files.
    '''

    __ctx = mx.gpu()

    def get_ctx(self):
        ''' getter '''
        return self.__ctx

    def set_ctx(self, value):
        ''' setter '''
        self.__ctx = value

    ctx = property(get_ctx, set_ctx)

    def draw(self):
        '''
        Draws samples from the `true` distribution.
        
        Returns:
            `np.ndarray` of samples.
        '''
        arr = super().draw()
        arr = nd.ndarray.array(arr, ctx=self.__ctx)
        return arr