torch.nn.functional.dropout

torch.nn.functional.dropout(input, p=0.5, training=True, inplace=False) [source]

Во время обучения случайным образом обнуляет некоторые элементы тензора-входа с вероятностью p используя выборки из распределения Бернулли.

См. Dropout для подробностей.

Параметры:

• p (float) – вероятность обнуления элемента. По умолчанию: 0.5
• training (bool) – применять дропаут, если True. По умолчанию: True
• inplace (bool) – Если установлено в True, операция будет выполнена на месте. По умолчанию: False

Тип возвращаемого значения:

Тензор