zz = torch.tensor([[1.,2.,3.],[4.,5.,6.]])
a = torch.sigmoid(zz)
a.shape
Out[10]: torch.Size([2, 3])
a
Out[11]:
tensor([[0.7311, 0.8808, 0.9526],
[0.9820, 0.9933, 0.9975]])
sigmoid的输出维度与输入矩阵维度一致,例如得到的第一个值0.7311,就是将输入矩阵中第一个数带到sigmoid中,1/(1+e**-1),e值按2.7182计算,结果就是0.7311,即sigmoid函数将矩阵中每一个数字映射到【0,1】之间,作为概率。