frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Split up detail.h (#7836)

Browse Source
This commit is contained in:
Peter Goldsborough
2018-05-30 08:55:34 -07:00
committed by GitHub
parent 15122e93bc
commit 4a80755834
39 changed files with 220 additions and 180 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/cpp/api/rnn.cpp
View File

@ -64,7 +64,7 @@ bool test_RNN_xor(Func&& model_maker, bool cuda = false) {
return true;
};
void check_lstm_sizes(variable_list tup) {
void check_lstm_sizes(std::vector<Variable> tup) {
// Expect the LSTM to have 64 outputs and 3 layers, with an input of batch
// 10 and 16 time steps (10 x 16 x n)