diff --git a/deeplearning4j/deeplearning4j-manifold/deeplearning4j-tsne/src/main/java/org/deeplearning4j/plot/Tsne.java b/deeplearning4j/deeplearning4j-manifold/deeplearning4j-tsne/src/main/java/org/deeplearning4j/plot/Tsne.java index 9efb88e24..c530d768a 100644 --- a/deeplearning4j/deeplearning4j-manifold/deeplearning4j-tsne/src/main/java/org/deeplearning4j/plot/Tsne.java +++ b/deeplearning4j/deeplearning4j-manifold/deeplearning4j-tsne/src/main/java/org/deeplearning4j/plot/Tsne.java @@ -108,7 +108,7 @@ public class Tsne { int n = X.rows(); // FIXME: this is wrong, another distribution required here - Y = randn(X.rows(), targetDimensions, Nd4j.getRandom()); + Y = Nd4j.randn(X.dataType(), X.rows(), targetDimensions); INDArray dY = Nd4j.zeros(n, targetDimensions); INDArray iY = Nd4j.zeros(n, targetDimensions); INDArray gains = Nd4j.ones(n, targetDimensions);