裁剪n维数组的最后2维

import numpy as np

# 创建一个4维数组
arr = np.array([[[[1, 2], [3, 4]], [[5, 6], [7, 8]]], [[[9, 10], [11, 12]], [[13, 14], [15, 16]]]])

# 裁剪最后2维
result = arr[..., 0:1, :]

print(result)

const tf = require('@tensorflow/tfjs-node');

// 创建一个4维张量
const tensor = tf.tensor([[[[1, 2], [3, 4]], [[5, 6], [7, 8]]], [[[9, 10], [11, 12]], [[13, 14], [15, 16]]]]);

// 裁剪最后2维
const result = tensor.slice([0, 0, 0, 0], [2, 2, 1, 2]);

result.array().then(data => console.log(data));

import org.nd4j.linalg.factory.Nd4j;

public class ArrayManipulation {
    public static void main(String[] args) {
        // 创建一个4维数组
        Nd4j.create(new double[][][][]{{{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}}, {{{9, 10}, {11, 12}}, {{13, 14}, {15, 16}}}});

        // 裁剪最后2维
        INDArray result = array.get(NDArrayIndex.all(), NDArrayIndex.all(), NDArrayIndex.interval(0, 1), NDArrayIndex.all());

        System.out.println(result);
    }
}