Method: Torch::NN::NLLLoss#initialize

Defined in:
lib/torch/nn/nll_loss.rb

#initialize(weight: nil, ignore_index: -100,, reduction: "mean") ⇒ NLLLoss

Returns a new instance of NLLLoss.



4
5
6
7
 
# File 'lib/torch/nn/nll_loss.rb', line 4

def initialize(weight: nil, ignore_index: -100, reduction: "mean")
  super(weight, reduction)
  @ignore_index = ignore_index
end