torche: quelle est la structure pour les tenseurs de différentes tailles?

Question

tableau? mais l'insertion de gros tenseurs sur une table n'est pas efficace, pas même possible à cause de la mémoire dans mon cas. Cela fonctionne parfaitement bien, mais il est laid:

local s_ = 0 
s_ = s_ + 1; local X_py_1 = fromfile(('%s/x_py_%.2f.bin'):format(data_dir, scales[s_])) 
s_ = s_ + 1; local X_py_2 = fromfile(('%s/x_py_%.2f.bin'):format(data_dir, scales[s_])) 
s_ = s_ + 1; local X_py_3 = fromfile(('%s/x_py_%.2f.bin'):format(data_dir, scales[s_])) 
s_ = s_ + 1; local X_py_4 = fromfile(('%s/x_py_%.2f.bin'):format(data_dir, scales[s_])) 
s_ = s_ + 1; local X_py_5 = fromfile(('%s/x_py_%.2f.bin'):format(data_dir, scales[s_])) 
s_ = s_ + 1; local X_py_6 = fromfile(('%s/x_py_%.2f.bin'):format(data_dir, scales[s_])) 
X_py = {X_py_1, X_py_2, X_py_3, X_py_4, X_py_5, X_py_6} 

Answer 1

montrer mon code .. s'il vous plaît vous aider à

X_py = {X_py_1, X_py_2, X_py_3, X_py_4, X_py_5, X_py_6} 
for i, v in ipairs(X_py) do 
    v = fromfile(('%s/x_py_%.2f.bin'):format(data_dir, scales[i-1])) 
end 

simplement en utilisant le tableau, le voulez-vous?

Answer 2

bien, cela fonctionne simplement:

X_py = {} 
for s, scale in ipairs(scales) do 
    X_py[s] = fromfile(('%s/x_py_%.2f.bin'):format(data_dir, scale)) 
end