火炬调整张量

时间:2015-04-03 15:05:26

标签: lua machine-learning resize torch

如何在火炬中调整Tensor的大小? 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]

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

1 个答案:

答案 0 :(得分:9)

你需要这样做:

images:resize(...)

你做了什么:

images.resize(...)

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

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