摘要: import torch import numpy as np x = torch.tensor([[1,2,3], [4,5,6], [7,8,9]]) y = torch.tensor([[3,2,1], [6,5,4], [9,8,7]]) z = x+y print(z) #直接使用加号就能 阅读全文
posted @ 2020-04-19 10:29 地球上最后一个直男 阅读(567) 评论(0) 推荐(0) 编辑
摘要: import torch a = torch.ones(5,3) #5*3的元素全为1的矩阵 print(a) b = torch.rand((5,3)) #5*3的元素为0-1之间的随机数的矩阵 print(b) c = torch.tensor([[1,2,3], [4,5,6], [7,8,9 阅读全文
posted @ 2020-04-19 10:01 地球上最后一个直男 阅读(4879) 评论(0) 推荐(0) 编辑