torch.flatten(t, start_dim=0, end_dim=-1) 的实现原理如下。假设类型为torch.tensor 的张量t 的形状如下所示:(2,4,3,5,6),则orch.flatten(t, 1, 3).shape 的结果 ...
確定! 回上一頁