espnet.nets.chainer_backend.transformer.subsampling.LinearSampling
Less than 1 minute
espnet.nets.chainer_backend.transformer.subsampling.LinearSampling
class espnet.nets.chainer_backend.transformer.subsampling.LinearSampling(idim, dims, dropout=0.1, initialW=None, initial_bias=None)
Bases: Chain
Linear 1D subsampling.
- Parameters:
- idim (int) – input dim
- odim (int) – output dim
- dropout_rate (flaot) – dropout rate
Initialize LinearSampling.
forward(xs, ilens)
Subsample x.
- Parameters:x (chainer.Variable) – input tensor
- Returns: subsampled x and mask