Table of Contents
Shortcuts

Source code for openrl.modules.networks.utils.vdn

import torch
import torch.nn as nn



[docs]class VDNBase(nn.Module):
    def __init__(self):
        super(VDNBase, self).__init__()


[docs]    def forward(self, agent_qs):
        return torch.sum(agent_qs, dim=1, keepdim=True)