Оба relu
, leakyrelu
имеют флаг inplace
, так почему бы и нет sigmoid
?
Signature: F.sigmoid(input)
F.relu(input, inplace=False)
Согласно документам:
nn.functional.sigmoid
is deprecated. Usetorch.sigmoid
instead.
Если вам нужна версия на месте, используйте sigmoid_
:
import torch
torch.manual_seed(0)
a = torch.randn(5)
print(a)
a.sigmoid_()
print(a)
tensor([ 1.5410, -0.2934, -2.1788, 0.5684, -1.0845])
tensor([0.8236, 0.4272, 0.1017, 0.6384, 0.2527])