import torch


def conv(input, weight):
    """
    Returns the convolution of input and weight tensors,
          where input contains sequential data.
          The convolution is along the sequence axis.
          input is of size [batchSize, inputDim, seqLength]
    """
    output = torch.nn.functional.conv1d(input=input, weight=weight)
    return output