pred = np.array(pred.data.cpu()[0])[0]

对于NumPy多维数组data而言，data[0,]，和data[0]、data[0,...]、data[0,:]一样，都表示数组0轴的第一个元素。比如，用一个三维数组data表示层、行、列的结构，那data[0,]、data[0]、data[0,...]、data[0,:]都表示第0层的行列，即一个二维数组。

>>> import numpy as np
>>> data = np.arange(24).reshape(2,3,4)
>>> data[0]
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> data[0,]
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> data[0,...]
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> data[0,:]
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])