需要注意GPU 上的Tensor 不能直接转换为NumPy ndarray,需要使用 .cpu() 先将GPU ... 得到tensor 的大小 print(pytorch_tensor1.shape) print(pytorch_tensor1.size())
確定! 回上一頁