Sha256: 9c14bad08e22fed4bddbce752a2b21cf8e9eca2596f8979441f92bf09fff9949
Contents?: true
Size: 302 Bytes
Versions: 56
Compression:
Stored size: 302 Bytes
Contents
module Torch
module NN
class BCELoss < WeightedLoss
def initialize(weight: nil, reduction: "mean")
super(weight, reduction)
end
def forward(input, target)
F.binary_cross_entropy(input, target, weight: @weight, reduction: @reduction)
end
end
end
end
Version data entries
56 entries across 56 versions & 1 rubygems