Sha256: ccf99ee480f88f30c4c3b0b579180828ba8dce80e67e72ca1a68078656df9056
Contents?: true
Size: 271 Bytes
Versions: 55
Compression:
Stored size: 271 Bytes
Contents
module Torch
module NN
class MultiLabelMarginLoss < Loss
def initialize(reduction: "mean")
super(reduction)
end
def forward(input, target)
F.multilabel_margin_loss(input, target, reduction: @reduction)
end
end
end
end
Version data entries
55 entries across 55 versions & 1 rubygems