def __init__(self, pad_shape, pad_value=None):
self.pad_shape = cde.TensorShape(pad_shape)
self.pad_value = cde.Tensor(
np.array(pad_value)) if pad_value is not None else pad_value
def __init__(self, pad_shape, pad_value=None):
if pad_value is not None:
pad_value = cde.Tensor(np.array(pad_value))
super().__init__(cde.TensorShape(pad_shape), pad_value)
def test_shape():
x = [2, 3]
s = cde.TensorShape(x)
assert s.as_list() == x
assert s.is_known()
