本文整理汇总了C++中tensor_type::order方法的典型用法代码示例。如果您正苦于以下问题:C++ tensor_type::order方法的具体用法?C++ tensor_type::order怎么用?C++ tensor_type::order使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tensor_type
的用法示例。
在下文中一共展示了tensor_type::order方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的C++代码示例。
示例1: operator
void operator ()(const tensor_type &delta, const tensor_type &x, tensor_type &x_gradient) {
CHECK_EQ(x.order(), delta.order());
// TODO(robertsdionne): Support arbitrary tensor order with an n-dimensional iterator.
for (auto i = 0; i < x.shape().at(0); ++i) {
for (auto j = 0; j < x.shape().at(1); ++j) {
x_gradient.set({i, j}, delta.at({i, j}) * (x.at({i, j}) > F(0)));
}
}
}
示例2: operator
void operator ()(const tensor_type &x, tensor_type &y) {
using std::max;
CHECK_EQ(x.order(), y.order());
// TODO(robertsdionne): Support arbitrary tensor order with an n-dimensional iterator.
for (auto i = 0; i < x.shape().at(0); ++i) {
for (auto j = 0; j < x.shape().at(1); ++j) {
y.set({i, j}, max(F(0), x.at({i, j})));
}
}
}