1import numpy as np 2from mindspore import context, Tensor 3from mindspore.ops import operations as P 4from deepconvnet import DeepConvNet 5 6 7if __name__ == '__main__': 8 context.set_context(mode=context.GRAPH_MODE, device_target="GPU") 9 model = DeepConvNet() 10 model.set_train(False) 11 u = Tensor(np.random.random((2, 32, 1, 4096)).astype(np.float32)) 12 y = model(u) 13 print(P.Shape()(y), y) 14