| 548 | // Return the number of elements in a mode |
| 549 | template <int... Is, class Engine, class Layout> |
| 550 | CUTE_HOST_DEVICE constexpr |
| 551 | auto |
| 552 | size(Tensor<Engine,Layout> const& tensor) |
| 553 | { |
| 554 | return size<Is...>(tensor.layout()); |
| 555 | } |
| 556 | |
| 557 | // Return the rank of a mode |
| 558 | template <int... Is, class Engine, class Layout> |