train_x = torch.tensor(Zs, device=output_device) train_x = train_x.repeat((YD, 1, 1)) # Format the training labels - reshape train_y = torch.vstack(
確定! 回上一頁