speechbrain.lobes.models.VanillaNN 模块

用于简单测试的普通神经网络。

作者 * Elena Rastorgueva 2020

摘要

类：

VanillaNN

一个简单的普通深度神经网络。

参考

class speechbrain.lobes.models.VanillaNN.VanillaNN(input_shape, activation=<class 'torch.nn.modules.activation.LeakyReLU'>, dnn_blocks=2, dnn_neurons=512)[source]

基础类: Sequential

一个简单的普通深度神经网络。

Parameters:

input_shape (tuple) – 输入张量的预期形状。
activation (torch class) – 用于构建激活层的类。
dnn_blocks (int) – 包含的线性神经块的数量。
dnn_neurons (int) – 线性层中的神经元数量。

Example

>>> inputs = torch.rand([10, 120, 60])
>>> model = VanillaNN(input_shape=inputs.shape)
>>> outputs = model(inputs)
>>> outputs.shape
torch.Size([10, 120, 512])