Edit: assumed_shape
Cancel