... :numwords].flatten().cpu() # torch.flatten() only exists from PyTorch 0.4.1 subvocab = closewordind[templateind, :numwords].view(-1).cpu() if addeos: ...
確定! 回上一頁