pytorch中的dropout在drop什么？

最近遇到了一个很基础的问题，就是pytorch中的dropout在面对一个n维的矩阵时，是会随机drop某一行、或者某一维上的一个向量，还是某一个元素呢？用试验稍微验证了下
import torch
m = torch.nn.Dropout(p=0.2)
input = torch.randn(5, 5)
output = m(input)

input为
tensor([[-0.2266, -0.6318, -0.1628, -0.0301, -1.2603],
[-2.3018, 0.7649, 1.3658, -0.6601, 0.1574],
[-0.7697, 0.1300, -1.9488, 0.9426, -0.2315],
[ 0.9873, 0.7713, 1.3725, -0.6127, 0.2403],
[

pytorch中的dropout在drop什么？最先出现在Python成神之路。