mirror of
https://github.com/explosion/spaCy.git
synced 2025-01-24 00:04:15 +03:00
Auto-format code with black (#10977)
Co-authored-by: explosion-bot <explosion-bot@users.noreply.github.com>
This commit is contained in:
parent
d50668dbf0
commit
6313787fb6
|
@ -23,7 +23,7 @@ def forward(model, X, is_train):
|
||||||
nI = model.get_dim("nI")
|
nI = model.get_dim("nI")
|
||||||
W = model.get_param("W")
|
W = model.get_param("W")
|
||||||
# Preallocate array for layer output, including padding.
|
# Preallocate array for layer output, including padding.
|
||||||
Yf = model.ops.alloc2f(X.shape[0] + 1, nF * nO * nP, zeros=False)
|
Yf = model.ops.alloc2f(X.shape[0] + 1, nF * nO * nP, zeros=False)
|
||||||
model.ops.gemm(X, W.reshape((nF * nO * nP, nI)), trans2=True, out=Yf[1:])
|
model.ops.gemm(X, W.reshape((nF * nO * nP, nI)), trans2=True, out=Yf[1:])
|
||||||
Yf = Yf.reshape((Yf.shape[0], nF, nO, nP))
|
Yf = Yf.reshape((Yf.shape[0], nF, nO, nP))
|
||||||
Yf[0] = model.get_param("pad")
|
Yf[0] = model.get_param("pad")
|
||||||
|
|
Loading…
Reference in New Issue
Block a user