qq_42363777的博客

pytorch中对Tensor进行分块的操作: split()tensor.split(selection_or_int,dim)，返回的是一个列表，第一个参数是Int, 则按照参数的值在dim上均分（最后一组保留）。如果是List, 则List中包含了每块的大小。reshape_bottleneck = torch.randn((8, 8, 16, 26, 26))t_fea_forward, _ = reshape_bottleneck.split([8 - 1, 1], dim=1) # n,

直接说结论，view()和flatten都是和输入共享内存的，flatten的好处是不用输入形状参数，直接指定维度，在这之后的都被拉平。view则是更加灵活示验证代码：import torcht = torch.tensor([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])# ---flatten用法----print('flatten1')print(to