在不考虑张量的dim情况下,参考文献【1】是个阅读不错的选择。考虑dim的情况下【2】可以参考。
torch.split(tensor, split_size_or_sections, dim=0)
参考文献
[1]PyTorch基础(八)----- torch.split()方法
[2]pytorch每日一学40(torch.split())将张量拆分为块(tensor的分割)
在不考虑张量的dim情况下,参考文献【1】是个阅读不错的选择。考虑dim的情况下【2】可以参考。
torch.split(tensor, split_size_or_sections, dim=0)
[1]PyTorch基础(八)----- torch.split()方法
[2]pytorch每日一学40(torch.split())将张量拆分为块(tensor的分割)