Method: Torch::NN::BCELoss#forward

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

#forward(input, target) ⇒ `Object`



8
9
10

# File 'lib/torch/nn/bce_loss.rb', line 8

def forward(input, target)
  F.binary_cross_entropy(input, target, weight: @weight, reduction: @reduction)
end