pytorch创建Tensor方法

a=t.Tensor([[1,2,3],[4,5,6]])
a.tolist()#把tensor转化为list
输出:[[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]

猜你喜欢

转载自blog.csdn.net/zouxiaolv/article/details/83098760