碼上歡樂
相關內容    簡體    繁體

torch.chunk()函數

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

 

將tensor按dim方向分割成chunks個tensor塊，返回的是一個元組。

示例：

 


免責聲明！

本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。



猜您在找 pytorch中torch.cat(),torch.chunk(),torch.split()函數的使用方法 pytorch中torch.chunk()方法 torch 中的損失函數 torch.clamp() 函數 torch.max()函數 torch topk函數 （原）torch的apply函數 torch常用的函數 torch.randn和torch.rand函數 【laravel5】詳解laravel的chunk和chunkById函數
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM