為您找到相關結果40個
pytorch torch.gather函數(shù)的使用_python_腳本之家
torch.gather 是 PyTorch 中用于在指定維度上通過索引從源張量中提取元素的函數(shù),它需要輸入張量、維度索引和索引張量,示例代碼展示了如何使用 torch.gather 從輸入張量中按索引提取元素,返回的結果張量形狀與索引張量相同+ 目錄pytorch torch.gather函數(shù)torch.gather 是PyTorch 中的一個用于從給定維度上按索引取值
www.dbjr.com.cn/python/3271096...htm 2025-5-29
pytorch人工智能之torch.gather算子用法示例_python_腳本之家
這篇文章主要介紹了pytorch人工智能之torch.gather算子用法示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪
www.dbjr.com.cn/article/2631...htm 2025-6-3
PyTorch中的nn.ConvTranspose2d模塊詳解_python_腳本之家
nn.ConvTranspose2d是PyTorch中用于實現(xiàn)二維轉置卷積的模塊,廣泛應用于生成對抗網(wǎng)絡(GANs)和卷積神經(jīng)網(wǎng)絡(CNNs)的解碼器中。該模塊通過參數(shù)如輸入輸出通道數(shù)、卷積核大小、步長、填充等,能控制輸出尺寸和避免棋盤效應
www.dbjr.com.cn/python/327106d...htm 2025-6-8
PyTorch之torch.matmul函數(shù)的使用及說明_python_腳本之家
PyTorch的torch.matmul是一個強大的矩陣乘法函數(shù),支持不同維度張量的乘法運算,包括廣播機制。提供了矩陣乘法的語法,參數(shù)說明,以及使用示例,幫助理解其應用方式和乘法規(guī)則
www.dbjr.com.cn/python/327103l...htm 2025-5-26
pytorch 同步機制的實現(xiàn)_python_腳本之家
在PyTorch中,當多個算子和內核被并行執(zhí)行時,PyTorch 通過 CUDA 的流和事件機制來管理并發(fā)和同步,本文就來介紹一下pytorch 同步機制,具有一定的參考價值,感興趣的可以了解一下
www.dbjr.com.cn/python/327774d...htm 2025-5-30