Sha256: 98fc02b0d9e3b7bfd614174c5946497167dd6c9290a65091c961ad77999a63b2
Contents?: true
Size: 244 Bytes
Versions: 56
Compression:
Stored size: 244 Bytes
Contents
module Torch
module NN
class KLDivLoss < Loss
def initialize(reduction: "mean")
super(reduction)
end
def forward(input, target)
F.kl_div(input, target, reduction: @reduction)
end
end
end
end
Version data entries
56 entries across 56 versions & 1 rubygems