[Compute] az vmss scale: Add new logic to support scaling VMSS in edge zone (#30704)

Author: Jing-song

src/azure-cli/azure/cli/command_modules/vm/custom.py

@@ -3940,6 +3940,8 @@ def scale_vmss(cmd, resource_group_name, vm_scale_set_name, new_capacity, no_wai
 
     vmss.sku.capacity = new_capacity
     vmss_new = VirtualMachineScaleSet(location=vmss.location, sku=vmss.sku)
+    if vmss.extended_location is not None:
+        vmss_new.extended_location = vmss.extended_location
     return sdk_no_wait(no_wait, client.virtual_machine_scale_sets.begin_create_or_update,
                        resource_group_name, vm_scale_set_name, vmss_new)