不红红黑路同

pytorch

c++(5) python(1) 未归档(6) 机器学习(2) 线性代数(1) 题解(50)

/ 注册

全部文章 / pytorch （共7篇）

Pytorch中nn.Embedding的原理及使用

参数1:vocab的大小参数2:d_model 参数3(padding_idx):可选参数，原张量中置为idx的词，都被映射成一个长为d_model的零向量例1 embedding=torch.nn.Embedding(10,3) input = torch.LongTensor([[1...

2022-02-14

0 530

PyTorch | tensor的创建和基本运算

一、创建 1.1 基本的创建函数 x1=torch.empty(3,2) x2=torch.rand(3,2) x3=torch.zeros(3,2) x4=torch.ones(3,2) print(x1) print(x2) print(x3) print(x4) 1.2 指定类型 ...

2022-02-14

0 279

PyTorch | torch和numpy之间的转换

目录 1 tensor->array(tensor.numpy()） 2 array->tensor(torch.from_numpy(array)) 1 tensor->array(tensor.numpy()） x=torch.ones(3,2) y=x.num...

2022-02-14

0 396

PyTorch | Autograd

一、requires_grad requires_grad是tensor类的一个属性，默认是false。可以在初始化时用requires_grad参数来确定，也可以通过tensor.requires_grad_(bool)函数来改变既有tensor是否需要自动求导。 x=torch.ones(3...

2022-02-14

0 249

PyTorch | 构建一个最简单的神经网络

一、定义网络 1.1 代码 # 导入若干工具包 import torch import torch.nn as nn import torch.nn.functional as F # 定义一个简单的网络类 class Net(nn.Module): def __init__(s...

2022-02-14

0 243

Pytorch | 训练模型计算损失时“log_softmax_lastdim_kernel_impl“ not implemented for ‘Long‘报错及解决

问题下面这段代码会报错："log_softmax_lastdim_kernel_impl" not implemented for 'Long' criterion = nn.CrossEntropyLoss() x=torch.tensor([[0,1,0,5]]) y=...

2022-02-14

0 1428

Pytorch | torch.max（）的用法

如果只输入一个张量作为参数，那么不管输入的张量是几维的，返回的都是一个数字张量，它是输入张量的最大值。 xx=torch.tensor([1,2,3,4]) print(torch.max(xx)) yy=torch.tensor([[1,2,3,4],[2,5,1,3]]) print(torc...

2022-02-14

0 396