eeva/tensorflow-haskell
1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-06-02 19:13:34 +02:00
Code Issues Releases Wiki Activity

Fix typo bringing variable names in line with comments

Browse Source
This commit is contained in:
Jonathan Kochems 2017-07-23 18:54:40 +01:00
parent 6b5dc0eed3
commit 9063b24db1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow-ops/src/TensorFlow/Gradient.hs
View File

@ -465,12 +465,12 @@ opGrad "Concat" nodedef _ix [dy]
x = map toT $ tail _ix
_i = toT $ head _ix
i = reshape _i one
n = length x
m = length x
s :: [Tensor Build Int32]
s = map shape x
di :: Tensor Build Int32
di = CoreOps.concat (scalar 0) $ map (\t -> CoreOps.slice t i one) s
dx = CoreOps.splitV (fromIntegral n) dy di _i
dx = CoreOps.splitV (fromIntegral m) dy di _i
reshapeZip = zipWith reshape
one = constant (Shape [1 :: Int64]) [1 :: Int32]