Pytorch中常用知识点总结

Pytorch中常用知识点总结
（1）torch.sigmoid与torch.nn.Sigmoid函数使用方法的区别
torch.sigmoid(input, *, out=None) → Tensor

注意对于torch.sigmoid函数来说，可以通过将张量作为输入直接放在括号里面。

torch.nn.Sigmoid介绍

import torch

m = torch.nn.Sigmoid()
input = torch.randn(2)
output = m(input)

print(output)

import torch
import torch.nn as nn

x = torch.randn(3)#x为1*3的张量，数值随机

print(x)
xs = nn.Sigmoid()(x)
print(xs)

注意nn.Sigmoid的使用方法是将输入的张量放在括号外面，若将xs=nn.Sigmoid()(x)表达式更改为xs=nn.Sigmoid(x)则会报错。