标签： nn

Python

Pytorch中torch.nn.Softmax的dim参数用法说明

2021年11月25日 2赞 31阅读 0评论

Pytorch中torch.nn.Softmax的dim参数使用含义涉及到多维tensor时，对softma…

Python

pytorch 中nn.Dropout的使用说明

2021年11月25日 0赞 28阅读 0评论

看代码吧~ Class USeDropout(nn.Module): def __init__(self): …

Python

浅谈pytorch中的nn.Sequential(*net[3: 5])是啥意思

2021年11月25日 1赞 34阅读 0评论

看到代码里面有这个 1 class ResNeXt101(nn.Module): 2 def __init__…