我想知道*在这种情况下是什么意思。我也看过文档,但是我仍然不太清楚这意味着什么。我在下面附加了一些上下文代码。
(顺便说一句,我还看到了它使用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