文章目录 1. 张量转换为Numpy数组 2 . Numpy 数组转换为张量 3. 张量和数字转换 1. 张量转换为Numpy数组 使用Tensor.numpy()函数可以将张量转换为ndarray数组,但是共享内存,可以使用copy()函数避免共享 data = torch.tensor([2,3,4])
print('data的类型:',type(data))
# 将张量转换为 numpy 数组
numpy_data = data.numpy()
print('numpy_data的类型:'