🌈个人主页:羽晨同学
💫个人格言:“成为自己未来的主人~”
张量转换为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())