当前位置：网站首页>【Pytorch基础】torch.split()用法

【Pytorch基础】torch.split()用法

2022-04-23 04:11:00 【一穷二白到年薪百万】

在不考虑张量的dim情况下，参考文献【1】是个阅读不错的选择。考虑dim的情况下【2】可以参考。

torch.split(tensor, split_size_or_sections, dim=0)

参考文献

[1]PyTorch基础（八）----- torch.split()方法
[2]pytorch每日一学40(torch.split())将张量拆分为块(tensor的分割)

版权声明
本文为[一穷二白到年薪百万]所创，转载请带上原文链接，感谢
https://blog.csdn.net/zfhsfdhdfajhsr/article/details/124341349

边栏推荐

猜你喜欢

随机推荐