Sha256: d40da3a0784f1ccf9ec19949e2b8366cc5b2ed103a9fb2a411b72820497cb300
Contents?: true
Size: 521 Bytes
Versions: 54
Compression:
Stored size: 521 Bytes
Contents
module Torch
module NN
class InstanceNorm1d < InstanceNorm
def _check_input_dim(input)
if input.dim == 2
raise ArgumentError,
"InstanceNorm1d returns 0-filled tensor to 2D tensor." +
"This is because InstanceNorm1d reshapes inputs to" +
"(1, N * C, ...) from (N, C,...) and this makes" +
"variances 0."
end
if input.dim != 3
raise "expected 3D input (got #{input.dim}D input)"
end
end
end
end
end
Version data entries
54 entries across 54 versions & 1 rubygems