We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9d354fd commit 0de614fCopy full SHA for 0de614f
test/gpt2/layer/tblock.m
@@ -37,7 +37,7 @@ function outputHasInputSizeWithPasts(test,Input)
37
% Provide a fake past of sequence length 1
38
K_fake = dlarray(rand(C,1));
39
V_fake = dlarray(rand(C,1));
40
- past = cat(4,K_fake,V_fake);
+ past = cat(5,K_fake,V_fake);
41
[y,present] = test.block(x,past,weights,hyperParameters);
42
test.verifySize(y,size(x));
43
% The size of presents is the size of past except the sequence
0 commit comments