为什么 PyTorch 的内置损失函数只适用于 Long 张量类型？

【问题标题】：Why does PyTorch's built-in loss function only work with the Long tensor type?为什么 PyTorch 的内置损失函数只适用于 Long 张量类型？
【发布时间】：2021-09-23 16:13:49
【问题描述】：

所以，我使用的是：torch.nn.CrossEntropyLoss(predictions, targets)，我想知道为什么目标需要是 64 位整数而不是 32 位？

【问题讨论】：

标签： pytorch

【解决方案1】：

我想知道为什么目标需要是 64 位整数而不是 32 位？

这是因为 PyTorch 是预编译的。它之前的一些其他框架（不是 Tensorflow）会即时调用编译器，这可能会导致延迟和其他一些不愉快。 PyTorch 没有。但这意味着开发人员必须注意预编译库的大小。他们必须在支持另一种数据类型的效用与为该数据类型编译所有内容所导致的大小增加之间取得平衡，而这里的决定与int32、uint32、uint64 等背道而驰。

【讨论】：