这个屌丝很懒，什么也没留下！

nn.sigmoid()

作者：2023面试高手 | 2024-03-07 13:44:34


import torch
import torch.nn as nn
a = torch.arange(4).reshape(2,2).float()
print('a',a)
b =nn.Sigmoid()
print(b(a))

这个nn.sigmoid()完成的是对位置内每一个都按公式计算


a tensor([[0., 1.],
        [2., 3.]])
tensor([[0.5000, 0.7311],
        [0.8808, 0.9526]])

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/2023面试高手/article/detail/205941