动手学习深度学习笔记——PyTorch版(一)
torch基本操作import torch# 创建x = torch.arange(12, dtype=torch.float32) # 创建值为0-11的向量y = torch.tensor([[1,2,3],[4,5,6]]) # 通过数组创建张量torch.zeros((2,3,4)) # 形状为 2*3*4 的全0数组torch.ones((2,3,4)) # 形状为 2*3*4 的全1数组# 操作x.shape # 查看张量
原创
2022-03-05 21:31:11 ·
939 阅读 ·
0 评论