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