Sha256: 2349fb5847a73fc554f83a51601aa30c8259c818cb62ac09cfcb3bb50131d263
Contents?: true
Size: 190 Bytes
Versions: 56
Compression:
Stored size: 190 Bytes
Contents
module Torch module NN class WeightedLoss < Loss def initialize(weight, reduction) super(reduction) register_buffer("weight", weight) end end end end
Version data entries
56 entries across 56 versions & 1 rubygems