devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support reshaping with static shapes and dimensions conversion in tensor broadcasting

Browse Source
This commit is contained in:
Eugene Zhulenev
2018-09-14 15:25:27 -07:00
parent 9b864cdb37
commit 1b8d70a22b
4 changed files with 35 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h
View File

@@ -641,7 +641,7 @@ struct TensorEvaluator<const TensorBroadcastingOp<Broadcast, ArgType>, Device>
return;
}
const Dimensions& input_dims = m_impl.dimensions();
const Dimensions& input_dims = Dimensions(m_impl.dimensions());
// Pre-fill input_block_sizes, broadcast_block_sizes,
// broadcast_block_strides, and broadcast_tensor_strides. Later on we will

10
unsupported/Eigen/CXX11/src/Tensor/TensorDimensions.h
View File

@@ -290,6 +290,16 @@ struct DSizes : array<DenseIndex, NumDims> {
}
}
#ifdef EIGEN_HAS_INDEX_LIST
EIGEN_DEVICE_FUNC
template <typename FirstType, typename... OtherTypes>
DSizes(const Eigen::IndexList<FirstType, OtherTypes...>& dimensions) {
for (int i = 0; i < dimensions.count; ++i) {
(*this)[i] = dimensions[i];
}
}
#endif
#ifndef EIGEN_EMULATE_CXX11_META_H
template <typename std::ptrdiff_t... Indices>
EIGEN_DEVICE_FUNC DSizes(const Sizes<Indices...>& a) {