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()