Pytorch remodeler tenseur de dimension

Par exemple, j'ai 1D d'un vecteur de dimension (5). Je voudrais modifier la forme en 2D matrice (1,5).

Voici comment je le fais avec numpy

>>> import numpy as np
>>> a = np.array([1,2,3,4,5])
>>> a.shape
(5,)
>>> a = np.reshape(a, (1,5))
>>> a.shape
(1, 5)
>>> a
array([[1, 2, 3, 4, 5]])
>>> 

Mais comment puis-je faire avec Pytorch Tenseur (et Variable). Je ne veux pas revenir à numpy et l'interrupteur de la Torche variable de nouveau, car il sera le perte les informations.

Voici ce que j'ai dans Pytorch

>>> import torch
>>> from torch.autograd import Variable
>>> a = torch.Tensor([1,2,3,4,5])
>>> a

 1
 2
 3
 4
 5
[torch.FloatTensor of size 5]

>>> a.size()
(5L,)
>>> a_var = variable(a)
>>> a_var = Variable(a)
>>> a_var.size()
(5L,)
.....do some calculation in forward function
>>> a_var.size()
(5L,)

Maintenant je le veux de taille (1, 5).
Comment puis-je redimensionner ou remodeler la dimension de pytorch tenseur dans la Variable sans perte de grad de l'information. (parce que je les ferai paître dans un autre modèle d'avant en arrière)

OriginalL'auteur Haha TTpro | 2017-04-10