Sha256: 2756f7bc5f02f94db16cebfb601ca25a5b02fc3ebf865f7e468ea05ece82d3f9
Contents?: true
Size: 242 Bytes
Versions: 54
Compression:
Stored size: 242 Bytes
Contents
module Torch
module NN
class InstanceNorm2d < InstanceNorm
def _check_input_dim(input)
if input.dim != 4
raise ArgumentError, "expected 4D input (got #{input.dim}D input)"
end
end
end
end
end
Version data entries
54 entries across 54 versions & 1 rubygems