`transform_out` has to account for that, yes. But there's no way to do things with the same transform for both A and B, if that's what you're getting at. Unless we make a network that does complex multiplication, but I don't think that's worth it.