tcn¶
-
class
TemporalBlock
(n_inputs, n_outputs, kernel_size, stride, dilation, padding, dropout=0.2)[源代码]¶ The TCN block, consists of dilated causal conv, relu and residual block. See the Figure 1(b) in https://arxiv.org/pdf/1803.01271.pdf for more details.
- 参数
n_inputs (int) -- The number of channels in the input tensor.
n_outputs (int) -- The number of filters.
kernel_size (int) -- The filter size.
stride (int) -- The stride size.
dilation (int) -- The dilation size.
padding (int) -- The size of zeros to be padded.
dropout (float, optional) -- Probability of dropout the units. Defaults to 0.2.