Caffe2填坑系列(5)----获取输入输出数据----CPU与GPU稍有不同

  • CPU
    const auto& a = Input(0)获取第一个输入给a,a的类型是Tensor
    auto* b = Output(0)获取第一个输入的指针赋给b
    获取指向Tensor内数据的指针:
    const auto* a1 = a.template data(),需要注意两点1.data()前的template ;2、输入一般是只读,我们不希望改变起内部数据,data()返回的是一个静态指针。
    auto* b1 = b->template mutable_data(),需要注意两点1、mutable_data()前的template;2、mutable_data(),代表的时可变数据,只用我们需要写输数据时才使用mutable_data()。
  • GPU
    获取数据的方式相同。不同点在于获取指向Tensor内数据的指针:
    const auto* a1 = a.data()
    auto* b1 = b.mutable_data()
    可以看出data()和mutable_data()前不需要template。
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值