Python sparse_repeat Examples

Example #1

File: test_sparse.py Project: yohanJung/gpytorch

    def test_sparse_repeat_1d(self):
        sparse_1d = sparse_getitem(self.sparse, 1)
        actual = sparse_1d.to_dense().repeat(3, 1)
        res = sparse_repeat(sparse_1d, 3, 1)
        self.assertTrue(torch.equal(actual, res.to_dense()))

        actual = sparse_1d.to_dense().repeat(2, 3)
        res = sparse_repeat(sparse_1d, 2, 3)
        self.assertTrue(torch.equal(actual, res.to_dense()))

Example #2

def test_sparse_repeat_1d():
    sparse_1d = sparse_getitem(sparse, 1)
    actual = sparse_1d.to_dense().repeat(3, 1)
    res = sparse_repeat(sparse_1d, 3, 1)
    assert torch.equal(actual, res.to_dense())

    actual = sparse_1d.to_dense().repeat(2, 3)
    res = sparse_repeat(sparse_1d, 2, 3)
    assert torch.equal(actual, res.to_dense())

Example #3

File: test_sparse.py Project: yohanJung/gpytorch

    def test_sparse_repeat_2d(self):
        actual = self.sparse.to_dense().repeat(3, 2)
        res = sparse_repeat(self.sparse, 3, 2)
        self.assertTrue(torch.equal(actual, res.to_dense()))

        actual = self.sparse.to_dense().repeat(1, 2)
        res = sparse_repeat(self.sparse, 1, 2)
        self.assertTrue(torch.equal(actual, res.to_dense()))

        actual = self.sparse.to_dense().repeat(3, 1)
        res = sparse_repeat(self.sparse, 3, 1)
        self.assertTrue(torch.equal(actual, res.to_dense()))

Example #4

def test_sparse_repeat_2d():
    actual = sparse.to_dense().repeat(3, 2)
    res = sparse_repeat(sparse, 3, 2)
    assert torch.equal(actual, res.to_dense())

    actual = sparse.to_dense().repeat(1, 2)
    res = sparse_repeat(sparse, 1, 2)
    assert torch.equal(actual, res.to_dense())

    actual = sparse.to_dense().repeat(3, 1)
    res = sparse_repeat(sparse, 3, 1)
    assert torch.equal(actual, res.to_dense())