torch.zeros https://pytorch.org/docs/stable/generated/torch.zeros.html torch.repeat https://pytorch.org/docs/stable/generated/torch.Tensor.repeat.html torch.unsqueeze https://pytorch.org/docs/stable/generated/torch.unsqueeze.html?highlight=unsqueeze