3D張量作爲火炬中的圖像

Question

如何將3D張量保存爲火炬中的圖像？

我用形象包裝：https://github.com/torch/image/blob/master/doc/saveload.md

爲三維張量保存爲圖像：

image.save('train100.jpg', trainData[100]) 

加載回來的時候：

x = image.load('train100.jpg'.jpg') 

我得到x是不同於原始張量和保存的圖像都是白色的。

這是代碼：

require 'torch' 
require 'image' 
local trainset = torch.load('cifar.torch/cifar10-train.t7') 
local testset = torch.load('cifar.torch/cifar10-test.t7') 

local trainData = trainset.data:float() 
local trainLabels = trainset.label:float():add(1) 

testData = testset.data:float() 
testLabels = testset.label:float():add(1) 

image.save('train100.jpg', trainData[100]) 

x = image.load('train100.jpg') 
print(trainData[100]) 
print(x) 

Answer 1

這是通過標準化的張量解：

image.save('train100.jpg', trainData[100])