bs = 64  # batch size

[xb](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor") = [x_train](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor")[0:bs]  # a mini-batch from x
[preds](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor") = [model](https://docs.pytorch.org/docs/stable/generated/torch.nn.Sequential.html#torch.nn.Sequential "torch.nn.Sequential")([xb](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor"))  # predictions
[preds](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor")[0], [preds.shape](https://docs.pytorch.org/docs/stable/size.html#torch.Size "torch.Size")
print([preds](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor")[0], [preds.shape](https://docs.pytorch.org/docs/stable/size.html#torch.Size "torch.Size"))