数据操作-转换为Numpy张量

print(type(X))

A = X.numpy()
print(type(A))

B = torch.tensor(A)
print(type(B))

print(id(X))
print(id(A))
print(id(B))

#输出结果

<class 'torch.Tensor'>
<class 'numpy.ndarray'>
<class 'torch.Tensor'>
2256463463360
2256464269744
2256464300544

可以发现,转换后的id值不同。即转换后的结果不共享内存。

二、将大小为1的张量转换为Python标量

可以调用 item 函数或 python 的内置函数

# 使用tensor给定元素的值

a = torch.tensor([3.5])

# a.iem()直接以3.5形式输出 a, a.item(), float(a), int(a) #输出结果 (tensor([3.5000]), 3.5, 3.5, 3)

  

posted @ 2021-07-23 10:04  小秦同学在上学  阅读(318)  评论(0编辑  收藏  举报