码上欢乐
相关内容   简体   繁体

torch.chunk()函数

本文转载自  查看原文  2020-12-01 22:41  1024    Pytorch/ torch.chunk()

 

将tensor按dim方向分割成chunks个tensor块，返回的是一个元组。

示例：

 


免责声明！

本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。



猜您在找 torch 中的损失函数 torch.clamp() 函数 torch.max()函数 torch topk函数 torch.randn和torch.rand函数 关于torch.norm函数的笔记 torch.randn_like()函数 torch torch.nn.MSELoss()函数解读 pytorch中torch.narrow()函数
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM