Pytorch系列-张量的类型转换

🌈个人主页：羽晨同学 

💫个人格言:“成为自己未来的主人~”   

张量转换为NumPy数组

使用Tensor.numpy()函数可以将张量转换为ndarray数组

# 1.将张量转换为numpy数组
data_tensor=torch.tensor([2,3,4])
# 使用张量对象中的numpy函数进行转换
data_numpy=data_tensor.numpy()
print(type(data_tensor))
print(type(data_numpy))

 NumPy数组转换成张量

使用from_numpy可以将ndarray数组转换为Tensor

使用torch.tensor可以将ndarray数组转换为Tensor

 使用from_numpy()可以将ndarray数组转换为Tensor

data_numpy=np.array([2,3,4])
# 将 numpy数组转换为张量类型
# 1.from_numpy
# 2. torch.tensor(ndarray)
data_tensor=torch.from_numpy(data_numpy)
print(data_tensor)
print(data_numpy)

使用torch.tensor()可以将ndarray数组转换为Tensor

data_numpy=np.array([2,3,4])
data_tensor=torch.tensor(data_numpy)
print(data_tensor)
print(data_numpy)

标量张量和数字转换

对于只有一个元素的张量，使用item()函数将该值从张量中提取出来

# 当张量只包含一个元素时，可以通过item()函数提取出该值
data=torch.tensor([30,])
print(data.item())
data=torch.tensor(30)
print(data.item())