Python _gen_mask示例

示例#1

文件： masked_autoregressive_test.py 项目： AbhinavJain13/tensorflow

 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)

示例#2

 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)