gpt4 book ai didi

lua - torch 调整张量大小

转载 作者:行者123 更新时间:2023-11-30 08:25:33 24 4
gpt4 key购买 nike

如何在 Torch 中调整张量的大小? https://github.com/torch/torch7/blob/master/doc/tensor.md#resizing 中记录的方法似乎不起作用。

images = image.load('image.png',1,'float')
print(images:size())
-- result: 224x224 [torch.LongStorage of size 2]

images.resize(torch.FloatTensor(224,224,1,1))
print(images:size())
-- result: 224x224 [torch.LongStorage of size 2]
-- expected: 224x224x1x1 [torch.LongStorage of size 4]

为什么这个方法不起作用?

最佳答案

你需要做:

images:resize(...)

你做了什么:

images.resize(...)

images.resize 不会将当前张量作为第一个参数传递。

images:resize(...) 相当于 images.resize(images, ...)

关于lua - torch 调整张量大小,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/29434729/

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