File size: 234 Bytes
37b9e99
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from torch import nn

class BaseInteraction(nn.Module):
    def __init__(self, **config):
        super().__init__()
        self.config = config

    def forward(self, hidden1, hidden2):
        NotImplementedError("no implemented")