Snake

constructor(channels: Int, name: String = "Snake", initAlpha: Tensor<T, V>? = null)(source)

Parameters

channels

Number of channels (for per-channel alpha)

name

Module name

initAlpha

Initial alpha parameter tensor (shape: channels)