python - 如何将 numpy.int64 分配给 torch.cuda.FloatTensor？

Question

我无法将 int64 分配给 torch 张量。我有以下张量

tempScale = torch.zeros((total, len(scale))).cuda() if useGpu else torch.zeros((nbPatchTotal, len(scale)))

在我的代码中，当我使用以下行时，它会抛出一条错误消息

tmpScale[:, j] = scale

错误信息是

类型错误:无法将 numpy.int64 分配给 torch.cuda.FloatTensor

我错过了什么？

Answer 1

在分配之前，您必须将 scale 转换为与 tmpScale 相同类型和设备的 torch 张量。

tmpScale[:, j] = torch.from_numpy(scale).to(tmpScale)

请注意，这是将 scale 从 int64 转换为 float32，这可能会导致 loss of precision如果 scale 中的值大于 2²⁴(大约 1600 万)。