2020年7月26日
关于pytorch中@和*的用处
摘要： 1、@是用来对tensor进行矩阵相乘的： import torch d = 2 n=50 X = torch.randn(n,d) true_w = torch.tensor([[-1.0],[2.0]]) y = X @ true_w + torch.randn(n,1)*0.1 print(X 阅读全文
posted @ 2020-07-26 23:29 九叶草 阅读(10807) 评论(0) 推荐(2) 编辑