zjplab

import torch.nn as nn
import torch

class SRU(nn.Module):
    """ Simple Recurrent Unit https://arxiv.org/pdf/1709.02755.pdf """

    def __init__(self, input_size, hidden_size, activation=F.tanh):
        super().__init__()
        self.input_size = input_size
        self.hidden_size = hidden_size

zjplab

// Use Gists to store code you would like to remember later on
console.log(window); // log the "window" object to the console