espnet.nets.pytorch_backend.tacotron2.cbhg.HighwayNet

source

class espnet.nets.pytorch_backend.tacotron2.cbhg.HighwayNet(idim)

Bases: Module

Highway Network module.

This is a module of Highway Network introduced in Highway Networks.

Initialize Highway Network module.

• Parameters:idim (int) – Dimension of the inputs.

forward(x)

Calculate forward propagation.

• Parameters:x (Tensor) – Batch of inputs (B, …, idim).
• Returns: Batch of outputs, which are the same shape as inputs (B, …, idim).
• Return type: Tensor