报错行:
padding_num = torch.zeros((1, sigma()[1]), dtype=float).to(device)
错误原因:
将()用作[]
修改后:
padding_num = torch.zeros([1, sigma()[1]]), dtype=float).to(device)
padding_num = torch.zeros((1, sigma()[1]), dtype=float).to(device)
将()用作[]
padding_num = torch.zeros([1, sigma()[1]]), dtype=float).to(device)