Datasets¶

class torchid.datasets.SubsequenceDataset(*tensors, subseq_len)[source]¶

A dataset returning sub-sequences extracted from longer sequences.

Parameters:: *tensors (Tensor) – tensors that have the same size of the first dimension.

Examples

>>> u = torch.randn(1000, 2) # 2 inputs
>>> y = torch.randn(1000, 3) # 3 outputs
>>> train_dataset = SubsequenceDataset(u, y, subseq_len=100)