小编Moh*_*dan的帖子

为什么 Dropout 不改变我的输入张量?

请参阅以下与输出相关的代码,

import torch
import torch.nn as nn
inputTensor = torch.tensor([1.0, 2.0, 3, 4, 5])

outplace_dropout = nn.Dropout(p=0.4)
print(inputTensor)
output_afterDropout = outplace_dropout(inputTensor)
print(output_afterDropout)
print(inputTensor)
Run Code Online (Sandbox Code Playgroud)

输出是:

tensor([1., 2., 3., 4., 5.])
tensor([1.6667, 3.3333, 0.0000, 6.6667, 0.0000])
tensor([1., 2., 3., 4., 5.])
Run Code Online (Sandbox Code Playgroud)

您能否详细说明为什么输入张量值仍然不变?

python pytorch dropout

2
推荐指数
1
解决办法
62
查看次数

标签 统计

dropout ×1

python ×1

pytorch ×1