Questions tagged «torch»

我对方法感到困惑 view()对以下代码片段中。 class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5) self.pool = nn.MaxPool2d(2,2) self.conv2 = nn.Conv2d(6, 16, 5) self.fc1 = nn.Linear(16*5*5, 120) self.fc2 = nn.Linear(120, 84) self.fc3 = nn.Linear(84, 10) def forward(self, x): x = self.pool(F.relu(self.conv1(x))) x = self.pool(F.relu(self.conv2(x))) x = x.view(-1, 16*5*5) x = F.relu(self.fc1(x)) x …

205 python  memory  pytorch  torch  tensor 

我正在阅读PyTorch的文档，并找到了他们编写的示例 gradients = torch.FloatTensor([0.1, 1.0, 0.0001]) y.backward(gradients) print(x.grad) 其中x是初始变量，从中构造y（3矢量）。问题是，梯度张量的0.1、1.0和0.0001参数是什么？该文档不是很清楚。

112 neural-network  gradient  pytorch  torch  gradient-descent 

我尝试完成安装过程 在MacOs Catalina上 Github的 第一步是在终端中执行： cd ~/ curl -s https://raw.githubusercontent.com/torch/ezinstall/master/install-deps | bash 但我得到错误： Error: caskroom/cask was moved. Tap homebrew/cask-cask instead. 然后我尝试了： brew tap caskroom/cask 但是我遇到了同样的错误。你有什么建议吗？

11 macos  homebrew  torch  macos-catalina  homebrew-cask