子模块

时间融合变压器的nn.Modules实现。

类

AddNorm(输入大小[, 跳过大小, 可训练加])	初始化内部模块状态，由 nn.Module 和 ScriptModule 共享。
GateAddNorm(输入大小[, 隐藏大小, ...])	初始化内部模块状态，由 nn.Module 和 ScriptModule 共享。
GatedLinearUnit(输入大小[, 隐藏大小, ...])	门控线性单元
GatedResidualNetwork(input_size, ...[, ...])	初始化内部模块状态，由 nn.Module 和 ScriptModule 共享。
InterpretableMultiHeadAttention(n_head, d_model)	初始化内部模块状态，由 nn.Module 和 ScriptModule 共享。
PositionalEncoder(d_model[, max_seq_len])	初始化内部模块状态，由 nn.Module 和 ScriptModule 共享。
ResampleNorm(输入大小[, 输出大小, ...])	初始化内部模块状态，由 nn.Module 和 ScriptModule 共享。
ScaledDotProductAttention([dropout, scale])	初始化内部模块状态，由 nn.Module 和 ScriptModule 共享。
TimeDistributed(模块[, batch_first])	初始化内部模块状态，由 nn.Module 和 ScriptModule 共享。
TimeDistributedInterpolation(输出大小[, ...])	初始化内部模块状态，由 nn.Module 和 ScriptModule 共享。
VariableSelectionNetwork(input_sizes, ...[, ...])	计算每个大小为 input_size 的 num_inputs 个变量的权重