我有一个张量：
t1 = torch.randn(564, 400)
我想将其展开到一个225600很长的一维张量。
我怎样才能做到这一点？
请注意Kris-From的文档字符串建议的view和之间的区别：reshapereshape
如果可能，返回的张量将是的视图
input。否则，它将是副本。连续输入和具有兼容步幅的输入可以重塑而无需复制...
因此，如果您的张量不是连续的，调用reshape应该处理使用过的张量将不得不处理的内容view；也就是说，调用t1.contiguous().view(...)以处理非连续的张量。
同样，可以使用faltten：t1 = t1.flatten()等效于view(-1)，它更具可读性。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句