如何展开 PyTorch 张量？

Question

I have a tensor:

t1 = torch.randn(564, 400)

I want to unroll it to a 1-d tensor that's 225600 long.

How can I do this?

Answer 1

Note the difference between view and reshape as suggested by Kris - From reshape 's docstring:

When possible, the returned tensor will be a view of input . Otherwise, it will be a copy. Contiguous inputs and inputs with compatible strides can be reshaped without copying...

So in case your tensor is not contiguous calling reshape should handle what one would have had to handle had one used view instead; That is, call t1.contiguous().view(...) to handle non-contiguous tensors.

Also, one could use faltten : t1 = t1.flatten() as an equivalent of view(-1) , which is more readable.

Answer 2

Pytorch 很像 numpy，所以你可以简单地做，

t1 = t1.view(-1) or t1 = t1.reshape(-1)