torch学习
WrittenByLight
这个作者很懒,什么都没留下…
展开
-
torch学习三之nn.Maxpool
torch学习三之nn.Maxpool1dMaxpool1DMaxpool2DMaxpool1D函数原型:torch.nn.MaxPool1d(kernel_size: Union[T, Tuple[T, ...]], stride: Optional[Union[T, Tuple[T, ...]]] = None, padding: Union[T, Tuple[T, ...]] = 0, dilation: Union[T, Tuple[T, ...]] = 1, return_indic原创 2020-11-25 18:31:26 · 1871 阅读 · 0 评论 -
torch学习二之nn.Convolution
torch学习二之nn.Conv1dnn.Conv1d函数参数输入数据维度转换关于kernelnn.Conv1d一维卷积通常用于处理文本数据函数参数首先看一下官网定义CLASS torch.nn.Conv1d(in_channels: int, out_channels: int, kernel_size: Union[T, Tuple[T]], stride: Union[T, Tuple[T]] = 1, padding: Union[T, Tuple[T]] = 0, dilation:原创 2020-11-25 12:31:54 · 303 阅读 · 0 评论 -
torch学习一之nn.embedding
torch学习nn.embeddingnn.embedding用法:#初始化self.word_embedding = nn.Embedding(词数,词的维数,padding_index)self.word_embedding(input_data)#加载预训练模型#way 1self.word_embedding.weight.data.copy_(pretraining_data)#way 2self.word_embedding = nn.Embedding.from_pret原创 2020-11-24 16:42:00 · 1527 阅读 · 0 评论