当前位置：网站首页>[pytoch foundation] torch Split() usage

[pytoch foundation] torch Split() usage

2022-04-23 04:36:00 【Poor to an annual salary of one million】

Without considering the tensor dim Under the circumstances , reference 【1】 It's a good choice to read . consider dim Under the circumstances 【2】 You can refer to .

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

reference

[1]PyTorch Basics （ 8、 ... and ）----- torch.split() Method
[2]pytorch Learn every day 40(torch.split()) Split the tensor into blocks (tensor Segmentation )

版权声明
本文为[Poor to an annual salary of one million]所创，转载请带上原文链接，感谢
https://yzsam.com/2022/04/202204230410588904.html

边栏推荐

猜你喜欢

随机推荐