nn.AvgPool1d expects a 3D input tensor in the shape [batch_size, seq_len, features] while you are trying to use a 2D tensor. Could you explain your use case a bit more and what result would be expected?