SoftMax.lua - github.com/torch/nn.git - Unnamed repository; edit this file 'description' to name the repository.

Welcome to mirror list, hosted at ThFree Co, Russian Federation.

SoftMax.lua - github.com/torch/nn.git - Unnamed repository; edit this file 'description' to name the repository.

summary refs log tree commit diff

path: root/SoftMax.lua

blob: 22f0edae8998558482296b2374415621b5f60589 (plain)

local SoftMax, _ = torch.class('nn.SoftMax', 'nn.Module')

function SoftMax:updateOutput(input)
   return input.nn.SoftMax_updateOutput(self, input)
end

function SoftMax:updateGradInput(input, gradOutput)
   return input.nn.SoftMax_updateGradInput(self, input, gradOutput)
end