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