import torch

from torch import nn

l = nn.Linear(4,4)
b = nn.BatchNorm1d(num_features = 4)
v = l.forward(torch.rand([4]))
bv = b(v)

This throws me ValueError: expected 2D or 3D input (got 1D input)
EDIT: [SOLVED] Nevermind the issue is because i didnt pass input in batches