Skip to contents

Depthwise separable convolution weight count

Source: R/geron_w4d_native.R
morie_geron_separable_params.Rd

Depthwise separable convolution weight count

Usage

morie_geron_separable_params(k, c_in, c_out)

Arguments

k

Kernel side.

c_in

Input channels.

c_out

Output channels.

Value

Integer parameter count k*k*c_in + c_in*c_out.

Examples

morie_geron_separable_params(k = 5L, c_in = c(1, 2, 3, 4, 5, 6, 7, 8),
  c_out = c(1, 2, 3, 4, 5, 6, 7, 8))
#> [1]  26  54  84 116 150 186 224 264