Sha256: 679843427c3050ded3389c215717ca4d81c902c2953e57d6aec009695186c2bc
Contents?: true
Size: 260 Bytes
Versions: 54
Compression:
Stored size: 260 Bytes
Contents
module Torch
module NN
class BatchNorm1d < BatchNorm
def _check_input_dim(input)
if input.dim != 2 && input.dim != 3
raise ArgumentError, "expected 2D or 3D input (got #{input.dim}D input)"
end
end
end
end
end
Version data entries
54 entries across 54 versions & 1 rubygems