Error line:
padding_num = torch.zeros((1, sigma()[1]), dtype=float).to(device)
wrong reason:
Use () as []
After modification:
padding_num = torch.zeros([1, sigma()[1]]), dtype=float).to(device)
padding_num = torch.zeros((1, sigma()[1]), dtype=float).to(device)
Use () as []
padding_num = torch.zeros([1, sigma()[1]]), dtype=float).to(device)