TL;DR: torch.flatten() Use torch.flatten() which was introduced in v0.4.1 and documented in v1.0rc1: >>> t = torch.tensor([[[1, 2], [3, 4]], [[5, 6], [7, ...
確定! 回上一頁