Source code for pfrl.policies.deterministic_policy
import torch
from torch import nn
from pfrl.distributions import Delta
[docs]class DeterministicHead(nn.Module):
"""Head module for a deterministic policy."""
def forward(self, loc):
return torch.distributions.Independent(Delta(loc=loc), 1)