Skip to main content
Search type Search syntax
Tags [tag]
Exact "words here"
Author user:1234
user:me (yours)
Score score:3 (3+)
score:0 (none)
Answers answers:3 (3+)
answers:0 (none)
isaccepted:yes
hasaccepted:no
inquestion:1234
Views views:250
Code code:"if (foo != bar)"
Sections title:apples
body:"apples oranges"
URL url:"*.example.com"
Saves in:saves
Status closed:yes
duplicate:no
migrated:no
wiki:no
Types is:question
is:answer
Exclude -[tag]
-apples
For more details on advanced search visit our help page
Results tagged with
Search options not deleted user 58261

Pytorch is an open source library for Tensors and Dynamic neural networks in Python with strong GPU acceleration. For details, see https://pytorch.org.

1 vote
2 answers
1k views

conv2d function in pytorch

I'm trying to use the function torch.conv2d from Pytorch but can't get a result I understand... … import pytorch filt = torch.rand(3, 3) im = torch.rand(3, 3) I want to compute a simple convolution with no padding, so the result should be a scalar (i.e. a 1x1 tensor). …
godot's user avatar
  • 113
0 votes

conv2d function in pytorch

import pytorch img = torch.rand(3, 3) model = torch.nn.Conv2d(in_channels=1, out_channels=1, kernel_size=(3, 3), stride=1, padding=0, bias=False) res = conv_mdl(img) print(res.shape) Which prints …
godot's user avatar
  • 113