来自Pytorch的辍学层更改未设置为零的值。使用Pytorch的文档示例:(source): import torch import torch.nn as nn m = nn.Dropout(p=0.5) input = torch.ones(5, 5)
確定! 回上一頁