import torch.nn.functional as F ... torch.cuda.set_device(1) # 这句用来设置pytorch在哪块GPU上运行 ... eps = torch.randn_like(std)
確定! 回上一頁