hxtorch.spiking.functional.dropout

Custom BatchDropout function

Functions

hxtorch.spiking.functional.dropout.batch_dropout(input: torch.Tensor, mask: torch.Tensor) → torch.Tensor

Applies a dropout mask to a batch of inputs.

Parameters

• input – The input tensor to apply dropout to.

• mask – The dropout mask. Entires in the mask which are False will disable their corresponding entry in input.

Returns

The input tensor with dropout mask applied.