relu
,leakyrelu
都带有inplace
标志,为什么不sigmoid
呢?
Signature: F.sigmoid(input)
F.relu(input, inplace=False)
答案 0 :(得分:1)
根据文档:
nn.functional.sigmoid
已过时。请改用torch.sigmoid
。
这里是sigmoid
的示例。我不确定它是否与就地操作相同,但是您可以尝试如下操作:
代码:
import torch
a = torch.randn(5)
print(a)
torch.sigmoid(a, out=a)
print(a)
输出:
tensor([-0.6651, 1.2416, -0.6720, 0.4406, -0.3336])
tensor([0.3396, 0.7758, 0.3381, 0.6084, 0.4174])