I have a loop in my DeepLearning Pipeline’s forward part to normalize the intermediate result.
Do it run on CPU and decrease the utilization of the GPU?
some snippet as follow:
def forward(self):
    ...
    for b in range(batch_size):
        self.points[b] = self.unit_cube(self.points[b])
    ....