Learning Soft Driving Constraints from Vectorized Scene Embeddings while Imitating Expert Trajectories