为什么在火炬的F.sigmoid中没有就位标记?

时间:2019-04-10 15:03:32

标签: pytorch

reluleakyrelu都带有inplace标志,为什么不sigmoid呢?

Signature: F.sigmoid(input)

F.relu(input, inplace=False)

1 个答案:

答案 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])

sigmoid docs