vollo_torch.nn

class vollo_torch.nn.PaddedConv1d(in_channels, out_channels, kernel_size, dilation=1, split_into=1)

Bases: torch.nn.modules.module.Module

torch.nn.Conv1d with left-padding such that the output sequence length is the same as the input sequence length

Can be split into multiple nodes in the compiler by setting split_into > 1

class vollo_torch.nn.LSTM(input_size, hidden_size, num_layers=1, bias=True, batch_first=False)

Bases: torch.nn.modules.module.Module

torch.nn.LSTM with zeros as the h_0 and c_0 initial states, and which discards the output h_n and c_n states.

class vollo_torch.nn.Ones

Bases: torch.nn.modules.module.Module

Instances of this class behaves similarly to torch.ones, but with the ability to make torch.fx not trace through it.

class vollo_torch.nn.Zeros

Bases: torch.nn.modules.module.Module

Instances of this class behaves similarly to torch.zeros, but with the ability to make torch.fx not trace through it.