PyTorch: get input layer size
I want to programatically find the size of my input layer.
If my first layer is called fc1, how do I find out its input?
pytorch
asked Nov 26 '18 at 6:10
Tom HaleTom Hale
7,7404965
add a comment |
I want to programatically find the size of my input layer.
If my first layer is called fc1, how do I find out its input?
pytorch
asked Nov 26 '18 at 6:10
Tom HaleTom Hale
7,7404965
add a comment |
I want to programatically find the size of my input layer.
If my first layer is called fc1, how do I find out its input?
pytorch
asked Nov 26 '18 at 6:10
Tom HaleTom Hale
7,7404965
I want to programatically find the size of my input layer.
If my first layer is called fc1, how do I find out its input?
pytorch
pytorch
asked Nov 26 '18 at 6:10
Tom HaleTom Hale
7,7404965
asked Nov 26 '18 at 6:10
Tom HaleTom Hale
7,7404965
edited Nov 26 '18 at 6:16
Tom Hale
asked Nov 26 '18 at 6:10
Tom HaleTom Hale
7,7404965
asked Nov 26 '18 at 6:10
Tom HaleTom Hale
7,7404965
asked Nov 26 '18 at 6:10
Tom HaleTom Hale
7,7404965
7,7404965
add a comment |
add a comment |
1 Answer
1
active
oldest
votes
Assuming your model is called model, this will give the number of input features of the fc1 layer:
model.fc1.in_features
This is useful inside the .forward() method:
def forward(self, x):
x = x.view(-1, self.fc1.in_features) # resize the input to match the input layer
...
answered Nov 26 '18 at 6:20
Tom HaleTom Hale
7,7404965
add a comment |
Your Answer
StackExchange.ifUsing("editor", function () {
StackExchange.using("externalEditor", function () {
StackExchange.using("snippets", function () {
StackExchange.snippets.init();
});
});
}, "code-snippets");
StackExchange.ready(function() {
var channelOptions = {
tags: "".split(" "),
id: "1"
};
initTagRenderer("".split(" "), "".split(" "), channelOptions);
StackExchange.using("externalEditor", function() {
// Have to fire editor after snippets, if snippets enabled
if (StackExchange.settings.snippets.snippetsEnabled) {
StackExchange.using("snippets", function() {
createEditor();
});
}
else {
createEditor();
}
});
function createEditor() {
StackExchange.prepareEditor({
heartbeatType: 'answer',
autoActivateHeartbeat: false,
convertImagesToLinks: true,
noModals: true,
showLowRepImageUploadWarning: true,
reputationToPostImages: 10,
bindNavPrevention: true,
postfix: "",
imageUploader: {
brandingHtml: "Powered by u003ca class="icon-imgur-white" href="https://imgur.com/"u003eu003c/au003e",
contentPolicyHtml: "User contributions licensed under u003ca href="https://creativecommons.org/licenses/by-sa/3.0/"u003ecc by-sa 3.0 with attribution requiredu003c/au003e u003ca href="https://stackoverflow.com/legal/content-policy"u003e(content policy)u003c/au003e",
allowUrls: true
},
onDemand: true,
discardSelector: ".discard-answer"
,immediatelyShowMarkdownHelp:true
});
}
});
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
StackExchange.ready(
function () {
StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fstackoverflow.com%2fquestions%2f53475580%2fpytorch-get-input-layer-size%23new-answer', 'question_page');
}
);
Post as a guest
Required, but never shown
1 Answer
1
active
oldest
votes
1 Answer
1
active
oldest
votes
active
oldest
votes
active
oldest
votes
Assuming your model is called model, this will give the number of input features of the fc1 layer:
model.fc1.in_features
This is useful inside the .forward() method:
def forward(self, x):
x = x.view(-1, self.fc1.in_features) # resize the input to match the input layer
...
answered Nov 26 '18 at 6:20
Tom HaleTom Hale
7,7404965
add a comment |
Assuming your model is called model, this will give the number of input features of the fc1 layer:
model.fc1.in_features
This is useful inside the .forward() method:
def forward(self, x):
x = x.view(-1, self.fc1.in_features) # resize the input to match the input layer
...
answered Nov 26 '18 at 6:20
Tom HaleTom Hale
7,7404965
add a comment |
Assuming your model is called model, this will give the number of input features of the fc1 layer:
model.fc1.in_features
This is useful inside the .forward() method:
def forward(self, x):
x = x.view(-1, self.fc1.in_features) # resize the input to match the input layer
...
answered Nov 26 '18 at 6:20
Tom HaleTom Hale
7,7404965
Assuming your model is called model, this will give the number of input features of the fc1 layer:
model.fc1.in_features
This is useful inside the .forward() method:
def forward(self, x):
x = x.view(-1, self.fc1.in_features) # resize the input to match the input layer
...
answered Nov 26 '18 at 6:20
Tom HaleTom Hale
7,7404965
answered Nov 26 '18 at 6:20
Tom HaleTom Hale
7,7404965
answered Nov 26 '18 at 6:20
Tom HaleTom Hale
7,7404965
answered Nov 26 '18 at 6:20
Tom HaleTom Hale
7,7404965
7,7404965
add a comment |
add a comment |
Thanks for contributing an answer to Stack Overflow!
- Please be sure to answer the question. Provide details and share your research!
But avoid …
- Asking for help, clarification, or responding to other answers.
- Making statements based on opinion; back them up with references or personal experience.
To learn more, see our tips on writing great answers.
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
StackExchange.ready(
function () {
StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fstackoverflow.com%2fquestions%2f53475580%2fpytorch-get-input-layer-size%23new-answer', 'question_page');
}
);
Post as a guest
Required, but never shown
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
