*在np数组shape的参数中是什么意思?例如np.multiply(* arr.shape)?

时间:2019-07-02 17:05:00

标签: python-3.x numpy

我想知道*在这种情况下是什么意思。我也看过文档,但是我仍然不太清楚这意味着什么。我在下面附加了一些上下文代码。

(顺便说一句,我还看到了它使用PyTorch张量的.view()或.reshape_()方法。想知道这是否是相同的意思。)

文档:https://docs.scipy.org/doc/numpy/reference/generated/numpy.multiply.html

def one_hot_encode(arr, n_labels):

    one_hot = np.zeros((np.multiply(*arr.shape), n_labels), dtype=np.float32)

    one_hot[np.arange(one_hot.shape[0]), arr.flatten()] = 1.

    one_hot = one_hot.reshape((*arr.shape, n_labels))

    return one_hot

0 个答案:

没有答案