Method test_multidim_bernoulli

test/encoding/test_encoding.py:20–27

(self)

                    assert spikes.size() == torch.Size((t, n))

    def test_multidim_bernoulli(self):
        for shape in [[5, 5], [10, 10], [25, 25]]:  # shape of nodes in layer
            for t in [1, 100]:  # number of timesteps
                for m in [0.1, 1.0]:  # maximum spiking probability
                    datum = torch.empty(shape).uniform_(0, m)
                    spikes = bernoulli(datum, time=t, max_prob=m)

                    assert spikes.size() == torch.Size((t, *shape))

    def test_bernoulli_loader(self):
        for s in [1, 100]:  # number of data samples