토치에서 딤섬 확장 방법 (텐서에 치수 추가)

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