我已经使用tf.Data（或者更准确地说是Pandasand的混合tf.Data）创建了一个管道。我的具体要求是使用window命令（它应用具有指定窗口大小、步长和步幅的移动窗口）创建序列，然后批处理这些窗口以训练时间序列 RNN。网络实际上是在JAX/中实现的，FLAX所以我使用as_numpy_iterator()从Tensor转换到np.array

我正在考虑迁移到 PyTorch 并希望迁移我的管道。PyTorch 中的等价物是什么？我找到了 torch.utils.data 但我没有看到任何关于窗口等的提及。