package lesson1; import org.nd4j.linalg.api.ndarray.INDArray; import org.nd4j.linalg.factory.Nd4j; /** * Created by Joe on 2017/12/27. * 获取并且设置数组部分 */ public class Nd4jGetAndSetParts { public static void main(String[] args) { INDArray nd = Nd4j.create(new float[]{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}, new int[]{2, 6}); System.out.println("原始数组"); System.out.println(nd); /* 获取一行 */ System.out.println("获取数组中的一行"); INDArray singleRow = nd.getRow(0); System.out.println(singleRow); /* 获取多行 */ System.out.println("获取数组中的多行"); INDArray multiRows = nd.getRows(0, 1); System.out.println(multiRows); /* 替换其中的一行 */ System.out.println("替换原有数组中的一行"); INDArray replaceRow = Nd4j.create(new float[]{1, 3, 5, 7, 9, 11}); nd.putRow(0, replaceRow); System.out.println(nd); } }