Pytorch
.我心永恒_
stay hungry, stay foolish. keep looking, don't sate!
展开
-
Pytorch之torch.nn.functional.pad函数详解
torch.nn.functional.pad函数是PyTorch内置的矩阵填充函数(1).tensor进行paddingtorch.nn.functional.pad(input, pad, mode,value ) Args: """ input:四维或者五维的tensor Variabe pad:不同Tensor的填充方式 1.四维Tensor:传入四元素tuple(pad_...原创 2019-08-28 18:20:16 · 43847 阅读 · 17 评论 -
Pytorch中的train和eval模式详解
(一)、model.train()和model.eval()分别在训练和测试中都要写,它们的作用如下:(1)model.train()启用 BatchNormalization 和 Dropout,将BatchNormalization和Dropout置为True(2).model.eval()不启用 BatchNormalization 和 Dropout,将BatchNormaliza...原创 2019-08-29 21:52:26 · 12797 阅读 · 2 评论 -
Pytorch中的scatter_函数
(1). scatter_函数详细描述如下:scatter_(input,dim,index,value) 将value对应的值按照index确定的索引写入本tensor中,其中索引是根据给定的dim(维度)来确定的,且index维度要与input维度一致"""Args:input:要进行scatter_填充的tensordim:要在input张量哪一个维度进行scatter_填充i...原创 2019-09-12 22:42:43 · 3055 阅读 · 1 评论