6 lines
201 B
Python
6 lines
201 B
Python
|
from .base_projection_layer import BaseProjectionLayer
|
||
|
|
|
||
|
|
class IdentityProjectionLayer(BaseProjectionLayer):
|
||
|
|
def project_from_rollouts(self, dist, rollout_data, **kwargs):
|
||
|
|
return dist, dist
|