def test346Inclusive(self): expected_mask = np.array([[1, 0, 0, 0], [1, 0, 0, 0], [1, 1, 0, 0], [1, 1, 0, 0], [1, 1, 1, 0], [1, 1, 1, 0]]) mask = masked_autoregressive._gen_mask(num_blocks=3, n_in=4, n_out=6, mask_type="inclusive") self.assertAllEqual(expected_mask, mask)
def test346Inclusive(self): expected_mask = np.array( [[1, 0, 0, 0], [1, 0, 0, 0], [1, 1, 0, 0], [1, 1, 0, 0], [1, 1, 1, 0], [1, 1, 1, 0]]) mask = _gen_mask(num_blocks=3, n_in=4, n_out=6, mask_type="inclusive") self.assertAllEqual(expected_mask, mask)
def test346Exclusive(self): expected_mask = np.array( [[0, 0, 0, 0], [0, 0, 0, 0], [1, 0, 0, 0], [1, 0, 0, 0], [1, 1, 0, 0], [1, 1, 0, 0]]) mask = _gen_mask(num_blocks=3, n_in=4, n_out=6, mask_type="exclusive") self.assertAllEqual(expected_mask, mask)