Newer version of PyTorch enforces strict device consistency for all tensors and indices#233

linhoangce wants to merge 2 commits intorasbt:mainrasbt/machine-learning-book:mainfrom

linhoangce:ch15-pt3-device-updatelinhoangce/machine-learning-book:ch15-pt3-device-updateCopy head branch name to clipboard