Sha256: 0c2c6f58c19de9d39f943c046ecbc40935873ec2effdfe65a11972ac3a680370
Contents?: true
Size: 177 Bytes
Versions: 51
Compression:
Stored size: 177 Bytes
Contents
module Torch module NN class AdaptiveAvgPool2d < AdaptiveAvgPoolNd def forward(input) F.adaptive_avg_pool2d(input, @output_size) end end end end
Version data entries
51 entries across 51 versions & 1 rubygems