pytorch基础入门day01

张开发
• 2026/4/9 4:55:09 • 15 分钟阅读

最新文章

推荐文章

相关文章

分享文章

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)

更多文章