pytorch基础入门day01
对pytorch的张量创建#张量:与numpy相似(tensor) # 分为维度,形状,数据类型 # 张量的创建 import torch # 创建一个2*3的全0张量 atorch.zeros(2,3) print(a) # one torch btorch.ones(2,3) # random torch ctorch.randn(2,3) # 从numpy中创建张量 import numpy as np numpy_arraynp.array([1,2],[3,4]) tensor_from_numpy torch.from_numpy(numpy_array) print(tensor_from_numpy) # 在指定的设备上创建张量 devicetorch.device(cuda if torch.cuda.is_available() else cpu) dtorch.randn(2,3,devicedevice) print(d)
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2498335.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!