mukNN 2022-04-27 16:58 采纳率: 100%
浏览 414
已结题

怎么将一个tensor里的数据都根据条件换成指定数据呢?

比如有一个大小1000的一维tensor数据,数据只有5种情况,0,1,2,3,4.
[0,1,2,3,4,2,3,2,4,1,2,0,1....]

现在给出一个5个数据的列表,比如 [10,1,-1,5,3]

希望tensor里的数据能替换,方法为:
比如tensor的第一个数据是0,那0就作为索引去找到 列表[0]的第一个数据,也就是10,将tensor的第一个数据从0换为10
也就是 tensor[x] = 列表[tensor[x]]

请问,有没有什么高效的替换方法,机器学习里面自定义损失函数时要用到所以希望运行效率高一些.

  • 写回答

1条回答 默认 最新

  • 不会长胖的斜杠 后端领域新星创作者 2022-04-27 17:09
    关注

    这样就可以了,望采纳!

    a = [0,1,2,3,4,2,3,2,4,1,2,0]
    b = [10,1,-1,5,3]
    a = list(map(lambda x: b[x], a))
    print(a)
    
    

    img

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 5月6日
  • 已采纳回答 4月28日
  • 创建了问题 4月27日

悬赏问题

  • ¥15 如何用Thoony写ESP32温湿度检测无源蜂鸣器报警代码?
  • ¥20 部件内部的CT图像数据集
  • ¥15 Visual studio调用动态库
  • ¥15 双目摄像头标定后的校准文件
  • ¥15 powerbi矩阵增加度量值后出现对应关系错乱
  • ¥30 频率分析法分析绘制奈奎斯特图、波特图
  • ¥15 弹出来一万个系统找不到指定的文件框框,怎么解决
  • ¥15 ADS生成的微带线为什么是蓝色空心的
  • ¥15 求一下解题思路,完全不懂
  • ¥15 tensorflow