Added support for tensor inflation.

This commit is contained in:
Benoit Steiner
2015-07-16 09:04:05 -07:00
parent b900fe47d5
commit 05787f8367
6 changed files with 308 additions and 0 deletions

View File

@@ -471,6 +471,11 @@ class TensorBase<Derived, ReadOnlyAccessors>
stride(const Strides& strides) const {
return TensorStridingOp<const Strides, const Derived>(derived(), strides);
}
template <typename Strides> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
const TensorInflationOp<const Strides, const Derived>
inflate(const Strides& strides) const {
return TensorInflationOp<const Strides, const Derived>(derived(), strides);
}
// Support for custom unary and binary operations
template <typename CustomUnaryFunc>