nd.array不支持字符串数组转型吗

nd.array不能转np.array(‘safasa’)的吗?

看了下代码, 目前并不支持 string 类型的 ndarray.
可以考虑去 github 提个 issue.

另外, 这里转化 string array 是要直接参与什么后续计算吗? 能否给个具体用例场景.

例如 我做多标签预测,那么最后(n, labels)形状的模型输出,计算损失的时候每个样本有自己的一个权重加权,那么我在Dataset的getItem()中,就需要返回每个样本的unique ID,然后通过这个ID找到对应的权重,现在的解决方法是通过sklearn的LableEncoder转换输出成int,因为DataLoader内部的Sampler没记错的话会将batch_size的样本数据进行ndarray转换。