exp_vml_cpu not implemented for 'Long'

最新推荐文章于 2024-06-27 17:27:12 发布

AI算法网奇

最新推荐文章于 2024-06-27 17:27:12 发布

阅读量2.6k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/104845214

pytorch知识宝典专栏收录该内容

512 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

遇到一个编程问题，'exp_vml_cpu' 函数不支持'Long'类型的数据，需要将数据类型转换为float32才能正常运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

exp_vml_cpu not implemented for 'Long'

from torch import nn

if __name__ == '__main__':

 
    import torch

    bce_loss = nn.BCEWithLogitsLoss(reduction='sum')
    labels=torch.Tensor([1,2])
    input_tensor = torch.tensor([[1, 2, 3],
                                 [4, 5,6]])

    index_num = torch.arange(0, labels.size(0))
    value, index = torch.max(input_tensor, dim=1)  # 按列SoftMax,列和为1

    # _, index = torch.topk(thetas, 1, dim=1, largest=True, sorted=True, out=None)

    select_pos = index != labels.long()

    pos_label = torch.ones((input_tensor[index_num, labels.long()][select_pos]).size(0))

    data=input_tensor[index_num, labels.long()][select_pos]
    pos_loss =  bce_loss(data, pos_label)

    print(pos_loss)

data需要是float32类型的，

pos_loss =  bce_loss(data.float(), pos_label)