gpt4 book ai didi

python - 如何使用滑动窗口调整 PyTorch 张量的大小?

转载 作者:行者123 更新时间:2023-12-04 10:36:33 25 4
gpt4 key购买 nike

我有一个大小为:torch.Size([118160, 1]) 的张量。我想要做的是将它分成 n 个张量,每个张量有 100 个元素,一次滑动 50 个元素。使用 PyTorch 实现这一目标的最佳方法是什么?

最佳答案

您可以使用 Pytorch 的展开 API。引用这个 https://pytorch.org/docs/stable/generated/torch.Tensor.unfold.html

例子:

x = torch.arange(1., 20)
x.unfold(0,4,2)

tensor([[ 1., 2., 3., 4.],
[ 3., 4., 5., 6.],
[ 5., 6., 7., 8.],
[ 7., 8., 9., 10.],
[ 9., 10., 11., 12.],
[11., 12., 13., 14.],
[13., 14., 15., 16.],
[15., 16., 17., 18.]])

关于python - 如何使用滑动窗口调整 PyTorch 张量的大小?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60157188/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com