mortonjt

import torch
from torch import nn
from torch.autograd import Variable

class AdaptiveSoftmax(nn.Module):
    def __init__(self, input_size, cutoff):
        super().__init__()
        
        self.input_size = input_size
        self.cutoff = cutoff