* Moving tensors to the right device. * Switch to gpu.medium
Attach a file by drag & drop or click to upload