分类：inplace字段

2021
07-09

对PyTorch中inplace字段的全面理解

例如torch.nn.ReLU(inplace=True)inplace=True表示进行原地操作，对上一层传递下来的tensor直接进行修改，如x=x+3；inplace=False表示新建一个变量存储操作结果，如y=x+3，x=y；inplace=True可以节省运算内存，不用多存储变量。补充：PyTorch中网络里面的inplace=True字段的意思在例如nn.LeakyReLU(inplace=True)中的inplace字段是什么意思呢？有什么用？inplace=True的意思是进行原地操作，例如x=x+5，对x就是一个原地操作，y=x+5... 继续阅读 >