| 771 | } |
| 772 | |
| 773 | __device__ CUMAT_STRONG_INLINE Scalar coeff(Index row, Index col, Index batch, Index index) const |
| 774 | { |
| 775 | return child_.derived().coeff( |
| 776 | _Row == Axis::Row ? row |
| 777 | : _Col == Axis::Row ? col |
| 778 | : _Batch == Axis::Row ? batch |
| 779 | : 0, |
| 780 | _Row == Axis::Column ? row |
| 781 | : _Col == Axis::Column ? col |
| 782 | : _Batch == Axis::Column ? batch |
| 783 | : 0, |
| 784 | _Row == Axis::Batch ? row |
| 785 | : _Col == Axis::Batch ? col |
| 786 | : _Batch == Axis::Batch ? batch |
| 787 | : 0, |
| 788 | -1); |
| 789 | } |
| 790 | }; |
| 791 | |
| 792 |