Sha256: 4518499d473883f94e64831a0a2e5c61763d30fbdbf9fe9c79bf077fe1dcf680
Contents?: true
Size: 233 Bytes
Versions: 56
Compression:
Stored size: 233 Bytes
Contents
module Torch
module NN
class Softmax2d < Module
def forward(input)
raise ArgumentError, "Softmax2d requires a 4D tensor as input" unless input.dim == 4
F.softmax(input, dim: 1)
end
end
end
end
Version data entries
56 entries across 56 versions & 1 rubygems