import mindspore
import mindspore.nn as nn
import numpy as np
from mindspore import Tensor

mindspore.set_context(mode=mindspore.context.GRAPH_MODE, device_target="Ascend", device_id=0, reserve_class_name_in_scope=False, deterministic="ON")

class SimpleNetwork(nn.Cell):
    def __init__(self, length):
        super(SimpleNetwork, self).__init__()
        self.length = length

    def construct(self, scores):
        return scores[:, self.length]

inputs = np.load(r"npy/test.npy")
scores = Tensor(inputs)
simple_network = SimpleNetwork(50362)
print(simple_network(scores))