nn.ReLU(inplace=True)
nn.ReLU(inplace=True)inplace=True,用输出的数据覆盖输入的数据;节省空间,此时两者共用内存;
import torch
from torch import nn as nn
m0 = nn.ReLU(inplace=True)
input = torch.randn(8)
print(input)
output = m0(input)
print(output)
print(input)
结果为:
nn.ReLU(inplace=True)inplace=True,用输出的数据覆盖输入的数据;节省空间,此时两者共用内存;
import torch
from torch import nn as nn
m0 = nn.ReLU(inplace=True)
input = torch.randn(8)
print(input)
output = m0(input)
print(output)
print(input)
结果为: