ResBlock

ResNet block based on "Deep Residual Learning for Image Recognition". Pass the input through the ResBlock layer. The paper link is https://arxiv.org/pdf/1512.03385.pdf.

model_resblock(n_freq = 128)

Arguments

n_freq

the number of bins in a spectrogram. (Default: 128)

Value

Tensor shape: (n_batch, n_freq, n_time)

Details

forward param: specgram (Tensor): the input sequence to the ResBlock layer (n_batch, n_freq, n_time).

Examples

if(torch::torch_is_installed()) {
resblock = model_resblock()
input = torch::torch_rand(10, 128, 512)  # a random spectrogram
output = resblock(input)  # shape: (10, 128, 512)
}