Skip to content

Instantly share code, notes, and snippets.

@taisukeoe

taisukeoe/projection.scala

Last active September 2, 2015 05:09
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save taisukeoe/f3e4c0f4382e964c0163 to your computer and use it in GitHub Desktop.
Save taisukeoe/f3e4c0f4382e964c0163 to your computer and use it in GitHub Desktop.
Download ZIP
Row wise operations got weired
Raw
projection.scala
scala> val ndArray = (1d to 8d by 1).mkNDArray(Array(2,2,2),NDOrdering.C)
ndArray: org.nd4j.linalg.api.ndarray.INDArray =
[[[1.00,2.00]
[3.00,4.00]]
[[5.00,6.00]
[7.00,8.00]]]
scala> ndArray.sliceP.map(s => s.rowP.map(r => r*r))
res12: org.nd4j.linalg.api.ndarray.INDArray =
[[[1.00,2.00]
[9.00,16.00]]
[[5.00,6.00]
[49.00,64.00]]]
scala> ndArray.sliceP.map(s => s.columnP.map(r => r*r))
res13: org.nd4j.linalg.api.ndarray.INDArray =
[[[1.00,4.00]
[9.00,16.00]]
[[25.00,36.00]
[49.00,64.00]]]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment