It works just the same way, except you don't need any k-points in the non-periodic direction. QuantumATK, like most codes, treats 2D materials as 3D as well, just with a large vacuum region between each sheet to avoid electrostatic or direct interactions.