トーチのディムを拡大する方法（テンソルにディメンションを追加する）

Question

numpy.expand_dimsのように、テンソルにディメンションを追加したいとします。どうしたらいいですか？

私は次のコードでそれを行うことができます：それは（np.expand_dims(a,0)と比較）私には複雑すぎると思われる

a = torch.Tensor({{1,2}, {3,4}, {5,6}}) --array with shape (3,2) 
a_size_table = a:size():totable() 
table.insert(a_size_table, 1, 1) -- adding `1` before first dimension 
a:reshape(torch.LongStorage(a_size_table)) -- returning array with shape (1,3,2) 

。それを行うより良い方法があるかもしれませんか？

Answer 1

nn.utils.addSingletonDimensionがあります。私はここで答えを見つけました https://www.bountysource.com/issues/38773074-convenience-function-to-add-a-singleton-dimension-to-a-torch-tensor