Адаптирующая функция softmax pytorch - PullRequest
Новая
       25

Адаптирующая функция softmax pytorch

0 голосов
/ 04 мая 2018

В настоящее время я изучаю функцию softmax и хотел бы адаптировать изначально реализованный метод для небольших тестов.

Я был в документации, но не было так много полезной информации о функции. Это реализация pytorch python : 

def __init__(self, dim=None):
    super(Softmax, self).__init__()
    self.dim = dim

def __setstate__(self, state):
    self.__dict__.update(state)
    if not hasattr(self, 'dim'):
        self.dim = None

def forward(self, input):
    return F.softmax(input, self.dim, _stacklevel=5)

Где я могу найти имплементацию F.softmax?

Одна из вещей, которые я хочу попробовать, например: softmax-soft softmax, описанный здесь: Soft-Margin Softmax для глубокой классификации

С чего бы лучше всего начать? Заранее спасибо!

...