![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
__Watson__
这个作者很懒,什么都没留下…
展开
-
【pytorch】训练网络通用架构
WHAT IS TORCH.NN REALLY?pytorch中model eval和torch no grad()的区别import torchimport torch.nn as nnimport torch.nn.functional as Fimport torch.optim as optimfrom torch.utils.data import TensorDatasetfrom torch.utils.data import DataLoader# 定义网络结构cla.原创 2020-12-29 21:04:49 · 425 阅读 · 1 评论 -
【pytorch】使用matplotlib.pyplot绘制tensor类型的曲线
x,y均为Tensorplt.plot(x, y)报错:‘Tensor‘ object has no attribute ‘ndim‘原因:plt.plot()函数期望其输入为numpy数组,而不是torch.tensor需要用.numpy()将Tensor转化为numpy类型plt.plot(x.numpy(), y.numpy())再次报错:TypeError: can’t convert CUDA tensor to numpy. Use Tensor.cpu() to cop原创 2020-12-29 15:24:55 · 7530 阅读 · 0 评论 -
【pytorch】矩阵乘法 mm bmm matmul mul @ * 总结
1. 二维矩阵乘法 torch.mm()torch.mm(mat1, mat2, out=None)其中mat1(nxm), mat2(mxd), 输出out(nxd)一般只用来计算两个二维矩阵的矩阵乘法,而且不支持broadcast操作。torch.mm(input, mat2, out=None) → Tensor#对矩阵imput和mat2执行矩阵乘法。 如果input为(n x m)张量,则mat2为(m x p)张量,out将为(n x p)张量。#官方提示此功能不广播。有关广播的矩原创 2020-12-24 15:30:49 · 1096 阅读 · 0 评论