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: 609b3530af7dd2392cd05490dd43f63d2da90348 (plain)

local SoftMax, parent = 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