torch_layer_segregator
