【PYTORCH】Expected object of type torch. xxxTensor but found type torch. cuda. xxxTensor（torch0.4.0）

  The above figure shows the whole content of the error , Among them, it is modified for us bug The useful part is

  We can see from the picture , What we want layer（x,mask） Second parameter of ”mask“ Should be torch.ByteTensor Type instead of current torch.cuda.ByteTensor type . So we should put mask The type of is forcibly converted to torch.ByteTensor. That is, by modifying mask Of type To complete , The specific operation is as follows ：

layer（x,mask.type(torch.ByteTensor)）.