Function Conv1d

usr/diff/net.py:47–50

(*args, **kwargs)

def Conv1d(*args, **kwargs):
    layer = nn.Conv1d(*args, **kwargs)
    nn.init.kaiming_normal_(layer.weight)
    return layer


@torch.jit.script