Is torch.nn.functional.avg_pool1d deterministic?

1

There is no note in the description of avg_pool1d on this page that says "may choose a non-deterministic algorithm, so this function is deterministic?

The function conv1d, which is similar to avg_pool1d, has its Note, so please tell me what avg_pool1d is doing.

Thank you!!

2

As described in the note for F.conv1d cudnn might non-deterministically pick a kernel or the kernel itself might yield non-deterministic results, which is not the case for the pooling layer.

1 Like