Sha256: 8b728958b21f1d4a04be7f7e108cd05ce4711e7319466ba8db33e8ee2afdd5ce
Contents?: true
Size: 220 Bytes
Versions: 56
Compression:
Stored size: 220 Bytes
Contents
module Torch module NN class LogSoftmax < Module def initialize(dim: nil) super() @dim = dim end def forward(input) F.log_softmax(input, @dim) end end end end
Version data entries
56 entries across 56 versions & 1 rubygems