Sha256: 1732548e04f6c57ce5b7a90ae809fa8fc849af29b3a2108e06bd4851e860bc08
Contents?: true
Size: 236 Bytes
Versions: 54
Compression:
Stored size: 236 Bytes
Contents
module Torch
module NN
class BatchNorm3d < BatchNorm
def _check_input_dim(input)
if input.dim != 5
raise ArgumentError, "expected 5D input (got #{input.dim}D input)"
end
end
end
end
end
Version data entries
54 entries across 54 versions & 1 rubygems