The most common/code efficient way of implementing conditional batch norm in PyTorch is to use a batch norm without weights and then applying weight and shift 699.
I once implemented it by redoing things 599 for SNGAN.
Best regards
Thomas