pytorch - 为什么在 pytorch 的 F.sigmoid 中没有 inplace 标志？

Question

relu、leakyrelu 都有inplace 标记，为什么sigmoid 没有？

Signature: F.sigmoid(input)

F.relu(input, inplace=False)

Answer 1

根据文档:

nn.functional.sigmoid is deprecated. Use torch.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])

sigmoid docs