File size: 83 Bytes
6bc0d51
 
 
 
1
2
3
4
import torch.nn as nn

def create_super_module(size):
    return nn.Linear(size, 1)