PyTorch创建张量方法全解
本文介绍PyTorch创建张量的三种方式,分别为直接创建、依据数值创建以及依据概率创建。
1.直接创建
1.1 使用数组创建
1.1.1 语法:
1.1.2 说明:
1.1.3 程序:
arr=np.ones((3,3))
print("arr的数据类型为:"+str(arr.dtype))
t=torch.tensor(arr)
print(t)
1.1.4运行结果:
1.2 使用numpy创建
1.2.1 语法:
1.2.2说明:
1.2.3 程序:
arr=np.array([[1,2.
原创
2020-07-31 19:59:33 ·
2675 阅读 ·
0 评论