pytorch
筱踏云
什么都不会的nlper
展开
-
pytorch 判断并替换 nan
判断是否有 nantorch.any(torch.isnan(a))利用 torch.where() 函数替换所有 nanwhere() 函数有三个输入值,第一个是判断条件,第二个是符合条件的设置值,第三个是不满足条件的设置值。a = torch.Tensor([[1, 2, np.nan], [2, np.nan, 4], [3, 4, 5]])a = torch.where(torch.isnan(a), torch.full_like(a, 0), a)...原创 2020-05-19 17:42:38 · 20875 阅读 · 1 评论 -
pytorch 损失函数
这里只整理 pytorch 中常见的损失函数,便于自己查阅。BCELoss二分类用的交叉熵,用的时候需要在之前加一层 Sigmoid 函数,因为是二分类的话,只有正例和反例,且两者的概率和为 1。所以离散的形式可以写成:loss(xi,yi)=−wi(yilogxi+(1−yi)log(1−xi))loss(x_i, y_i) = -w_i(y_i\log{x_i} + (1-y_i)\...原创 2019-11-28 17:24:53 · 127 阅读 · 0 评论 -
pytorch Tensor 与 tensor
TensorTensor 是 pytorch 的一个class类,不同的数据类型对应不同的 Tensor 类,一共有九种:原创 2019-10-20 11:51:36 · 237 阅读 · 0 评论