非线性激活处理图片 sigmoid

import torch
import torchvision
from torch import nn
from torch.nn import ReLU
from torch.nn import Sigmoid


from torch.nn import MaxPool2d
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter

input = torch.tensor([
    [1,2,0,3,1],
    [0,1,2,3,1],
    [1,2,1,0,0],
    [5,2,3,1,1],
    [2,1,0,1,1]
],dtype=torch.float32)

input = torch.reshape(input,(-1,1,5,5))
print(input.shape)


data = torchvision.datasets.CIFAR10("./dataset",train=False,transform=torchvision.transforms.ToTensor())
dataloader = DataLoader(data,batch_size=64)


class Tudui(nn.Sigmoid):
    def __init__(self):
        super(Tudui, self).__init__()
        self.sigmoid1 = Sigmoid()
        # self.relu1 = ReLU()

    def forward(self,input):
        output = self.sigmoid1(input)
        # output = self.relu1(input)
        return output

writer=SummaryWriter("非线性激活日志")
step = 0
for data in dataloader:
    img,target = data
    writer.add_images("input",img,step)
    tudui = Tudui()
    output = tudui(img)
    writer.add_images("output",output,step)
    step = step+1
writer.close()   #cmd输入    tensorboard --logdir=logs   logs改成绝对路径