- 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。
Caffe2填坑系列(5)----获取输入输出数据----CPU与GPU稍有不同
最新推荐文章于 2020-03-04 19:05:09 发布