首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

PyTorch入门笔记-分割split函数

split torch.split(input, split_size_or_sections, dim = 0) 函数会将输入张量(input)沿着指定维度(dim)分割成特定数量的张量块,并返回元素为张量块的元素...tensor(Tensor)- 待分割的输入张量,此处的 tensor 参数和 torch.chunk 函数中的 input 参数类似,只需要注意使用关键字参数时候的参数名 split_size_or_sections...指定为 int 时 当传入 torch.split 函数中的 split_size_or_sections 参数为整型时(int),torch.split 函数和 torch.chunk 函数所实现的功能一样...,torch.split 函数中的 split_size_or_sections 参数和 torch.chunk 函数中的 chunks 参数等价。...使用 torch.split 函数,只需要为 split_size_or_sections 参数传入 [1, 2] 列表即可。

7.6K10

使用--gc-section编译选项减小程序体积

这种功能,讲道理编译器肯定要支持的,于是搜了一下,果然是有个编译选项 需要在编译的时候,加入-ffunction-sections, -fdata-sections 选项,在链接的时候,加入--gc-sections...选项 大概的意思就是,编译的时候,把每个函数作为一个section,每个数据(应该是指全局变量之类的吧)也作为一个section,这样链接的时候,--gc-sections会把没用到的section丢弃掉...如果不加-ffunction-sections选项,则默认似乎是每个源文件为一个section进行链接,这样子只要这个文件中用到了一个函数,那么所有的函数都会被链接进来。...所以编译器认为所有的函数都没用,于是全部丢弃了 赶紧试试,果然, 在lds文件中加上 ENTRY(_start) 之后,就好了,编译出来查看map文件,确实把没用到的函数丢弃了 但还有一个问题,就是不能加上 -fdata-sections...,加上就报错,只能用-ffunction-sections,这个暂时没解决 想起以前用STM32的时候,keil(MDK)就提供了一个选项,Options->C/C++ 中勾选上 “One ELF

1.5K30
领券