pytorch笔记(四)nn.Conv1d、nn.Conv2d、nn.Conv3d
生活随笔
收集整理的這篇文章主要介紹了
pytorch笔记(四)nn.Conv1d、nn.Conv2d、nn.Conv3d
小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
概念:
nn.Conv1d:常用在文本
- (B,C,L)
- (batch,channel,sequence_len)
- (批數(shù)量,通道數(shù),句子長(zhǎng)度)
nn.Conv2d:常用在圖像
- (B,C,H,W)
- (batch,channel,height,width)
- (批數(shù)量,通道數(shù),高度,長(zhǎng)度)
nn.Conv3d:常用在(時(shí)序)幀圖像
- (B,C,D,H,W)
- (batch,channel,depth,height,width)
- (批數(shù)量,通道數(shù),深度,高度,長(zhǎng)度)
一、參數(shù)解析
ps: * 號(hào)常用
| * in_channels(int) | 輸入信號(hào)的通道數(shù), 文本應(yīng)用中為詞向量的維度 | - |
總結(jié)
以上是生活随笔為你收集整理的pytorch笔记(四)nn.Conv1d、nn.Conv2d、nn.Conv3d的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。