10 #ifndef EIGEN_CXX11_TENSOR_TENSOR_BASE_H
11 #define EIGEN_CXX11_TENSOR_TENSOR_BASE_H
28 #ifndef EIGEN_PARSED_BY_DOXYGEN
31 template<
typename Derived>
62 random(
const RandomGenerator& gen = RandomGenerator())
const {
63 return nullaryExpr(gen);
216 igamma(
const OtherDerived& other)
const {
244 zeta(
const OtherDerived& other)
const {
335 template<
typename ScalarExponent>
338 pow(ScalarExponent exponent)
const
415 template <
int NanPropagation=PropagateFast>
419 return cwiseMax<NanPropagation>(constant(threshold));
422 template <
int NanPropagation=PropagateFast>
426 return cwiseMin<NanPropagation>(constant(threshold));
429 template<
typename NewType>
635 select(
const ThenDerived& thenTensor,
const ElseDerived& elseTensor)
const {
644 contract(
const OtherDerived& other,
const Dimensions& dims)
const {
650 contract(
const OtherDerived& other,
const Dimensions& dims,
const OutputKernel& output_kernel)
const {
657 convolve(
const KernelDerived& kernel,
const Dimensions& dims)
const {
664 fft(
const FFT& dims)
const {
683 template <
typename Reducer>
686 scan(
const Index& axis,
const Reducer& reducer,
bool exclusive =
false)
const {
693 sum(
const Dims& dims)
const {
733 template <
int NanPropagation=PropagateFast>
746 template <
int NanPropagation=PropagateFast>
755 all(
const Dims& dims)
const {
768 any(
const Dims& dims)
const {
811 in_dims[0] = return_dim;
824 in_dims[0] = return_dim;
833 reduce(
const Dims& dims,
const Reducer& reducer)
const {
870 const Index row_stride = 1,
const Index col_stride = 1,
871 const Index in_row_stride = 1,
const Index in_col_stride = 1,
874 in_row_stride, in_col_stride, 1, 1, padding_type, padding_value);
880 const Index row_stride,
const Index col_stride,
881 const Index in_row_stride,
const Index in_col_stride,
882 const Index row_inflate_stride,
const Index col_inflate_stride,
883 const Index padding_top,
const Index padding_bottom,
884 const Index padding_left,
const Index padding_right,
885 const Scalar padding_value)
const {
887 in_row_stride, in_col_stride, row_inflate_stride, col_inflate_stride,
888 padding_top, padding_bottom, padding_left, padding_right, padding_value);
894 const Index plane_stride = 1,
const Index row_stride = 1,
const Index col_stride = 1,
896 return TensorVolumePatchOp<Dynamic, Dynamic, Dynamic, const Derived>(
derived(), patch_planes, patch_rows, patch_cols, plane_stride, row_stride, col_stride, 1, 1, 1, 1, 1, 1, padding_type, padding_value);
903 const Index plane_stride,
const Index row_stride,
const Index col_stride,
904 const Index plane_inflate_stride,
const Index row_inflate_stride,
const Index col_inflate_stride,
905 const Index padding_top_z,
const Index padding_bottom_z,
906 const Index padding_top,
const Index padding_bottom,
908 return TensorVolumePatchOp<Dynamic, Dynamic, Dynamic, const Derived>(
derived(), patch_planes, patch_rows, patch_cols, plane_stride, row_stride, col_stride, 1, 1, 1, plane_inflate_stride, row_inflate_stride, col_inflate_stride, padding_top_z, padding_bottom_z, padding_top, padding_bottom, padding_left, padding_right, padding_value);
919 reshape(
const NewDimensions& newDimensions)
const {
929 stridedSlice(
const StartIndices& startIndices,
const StopIndices& stopIndices,
const Strides&
strides)
const {
951 roll(
const Rolls& rolls)
const {
956 pad(
const PaddingDimensions& padding)
const {
961 pad(
const PaddingDimensions& padding,
const Scalar padding_value)
const {
988 template <
typename CustomUnaryFunc>
993 template <
typename OtherDerived,
typename CustomBinaryFunc>
1006 template<
typename Format>
1011 #ifdef EIGEN_READONLY_TENSORBASE_PLUGIN
1012 #include EIGEN_READONLY_TENSORBASE_PLUGIN
1019 template <
typename Scalar,
int NumIndices,
int Options,
typename IndexType>
friend class Tensor;
1020 template <
typename Scalar,
typename Dimensions,
int Option,
typename IndexTypes>
friend class TensorFixedSize;
1035 template <
typename Scalar,
int NumIndices,
int Options,
typename IndexType>
friend class Tensor;
1036 template <
typename Scalar,
typename Dimensions,
int Option,
typename IndexTypes>
friend class TensorFixedSize;
1050 return derived() = this->random();
1054 return derived() = this->
template random<RandomGenerator>();
1061 internal::initialize_tensor<Derived, NumDimensions>(
eval, vals);
1106 reshape(
const NewDimensions& newDimensions)
const {
1128 stridedSlice(
const StartIndices& startIndices,
const StopIndices& stopIndices,
const Strides&
strides)
const {
1208 template <
typename DeviceType>
1214 template <
typename DeviceType,
typename DoneCallback>
1224 #ifdef EIGEN_TENSORBASE_PLUGIN
1225 #include EIGEN_TENSORBASE_PLUGIN
1236 Assign assign(
derived(), other.derived());
constexpr bool operator<=(const Log rhs, const Log lhs)
Internally used to filter on loglevel. Do not edit, as this is required for an optimised logger.
Definition: Logger.h:119
#define EIGEN_DEFAULT_COPY_CONSTRUCTOR(CLASS)
Macro to explicitly define the default copy constructor. This is necessary, because the implicit defi...
Definition: Macros.h:1119
#define EIGEN_DEFAULT_EMPTY_CONSTRUCTOR_AND_DESTRUCTOR(Derived)
Macro to manually define default constructors and destructors. This is necessary when the copy constr...
Definition: Macros.h:1137
#define EIGEN_DEVICE_FUNC
Definition: Macros.h:892
#define EIGEN_STRONG_INLINE
Definition: Macros.h:834
#define EIGEN_STATIC_ASSERT(X, MSG)
Definition: StaticAssert.h:26
SCALAR Scalar
Definition: bench_gemm.cpp:45
bool operator<(const benchmark_t &b1, const benchmark_t &b2)
Definition: benchmark-blocking-sizes.cpp:127
boost::multiprecision::number< boost::multiprecision::cpp_dec_float< 100 >, boost::multiprecision::et_on > Real
Definition: boostmultiprec.cpp:77
Definition: TensorAssign.h:57
Pseudo expression providing an operator = that will evaluate its argument asynchronously on the speci...
Definition: TensorDevice.h:88
Definition: TensorBase.h:33
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_cmp_op< Scalar, Scalar, internal::cmp_EQ >, const Derived, const TensorCwiseNullaryOp< internal::scalar_constant_op< Scalar >, const Derived > > operator==(Scalar threshold) const
Definition: TensorBase.h:606
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseNullaryOp< CustomNullaryOp, const Derived > nullaryExpr(const CustomNullaryOp &func) const
Definition: TensorBase.h:44
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp< internal::AndReducer, const Dims, const std::conditional_t< internal::is_same< bool, CoeffReturnType >::value, Derived, TensorConversionOp< bool, const Derived > > > all(const Dims &dims) const
Definition: TensorBase.h:755
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_polygamma_op< Scalar >, const Derived, const OtherDerived > polygamma(const OtherDerived &other) const
Definition: TensorBase.h:251
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_abs_op< Scalar >, const Derived > abs() const
Definition: TensorBase.h:311
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp< internal::MeanReducer< CoeffReturnType >, const Dims, const Derived > mean(const Dims &dims) const
Definition: TensorBase.h:705
const TensorReductionOp< internal::SumReducer< CoeffReturnType >, const DimensionList< Index, NumDimensions >, const Derived > sum() const
Definition: TensorBase.h:698
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_arg_op< Scalar >, const Derived > arg() const
Definition: TensorBase.h:317
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorChippingOp< DimId, const Derived > chip(const Index offset) const
Definition: TensorBase.h:935
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_imag_op< Scalar >, const Derived > imag() const
Definition: TensorBase.h:351
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorTraceOp< const Dims, const Derived > trace(const Dims &dims) const
Definition: TensorBase.h:839
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_zeta_op< Scalar >, const Derived, const OtherDerived > zeta(const OtherDerived &other) const
Definition: TensorBase.h:244
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorPairReducerOp< internal::ArgMinPairReducer< Pair< Index, CoeffReturnType > >, const array< Index, NumDimensions >, const Derived > argmin() const
Definition: TensorBase.h:796
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorLayoutSwapOp< const Derived > swap_layout() const
Definition: TensorBase.h:914
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp< internal::MaxReducer< CoeffReturnType, NanPropagation >, const Dims, const Derived > maximum(const Dims &dims) const
Definition: TensorBase.h:729
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_opposite_op< Scalar >, const Derived > operator-() const
Definition: TensorBase.h:83
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp< Reducer, const Dims, const Derived > reduce(const Dims &dims, const Reducer &reducer) const
Definition: TensorBase.h:833
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_ndtri_op< Scalar >, const Derived > ndtri() const
Definition: TensorBase.h:269
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_j0_op< Scalar >, const Derived > bessel_j0() const
Definition: TensorBase.h:167
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorScanProdOp cumprod(const Index &axis, bool exclusive=false) const
Definition: TensorBase.h:679
std::remove_const_t< Scalar > CoeffReturnType
Definition: TensorBase.h:38
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_log2_op< Scalar >, const Derived > log2() const
Definition: TensorBase.h:305
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseNullaryOp< internal::UniformRandomGenerator< Scalar >, const Derived > random() const
Definition: TensorBase.h:57
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_round_op< Scalar >, const Derived > round() const
Definition: TensorBase.h:440
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_sqrt_op< Scalar >, const Derived > sqrt() const
Definition: TensorBase.h:89
internal::traits< Derived > DerivedTraits
Definition: TensorBase.h:35
Eigen::IndexPair< Index > DimensionPair
Definition: TensorBase.h:640
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp< internal::MinReducer< CoeffReturnType, NanPropagation >, const Dims, const Derived > minimum(const Dims &dims) const
Definition: TensorBase.h:742
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_max_op< Scalar, Scalar, NaNPropagation >, const Derived, const OtherDerived > cwiseMax(const OtherDerived &other) const
Definition: TensorBase.h:496
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorSelectOp< const Derived, const ThenDerived, const ElseDerived > select(const ThenDerived &thenTensor, const ElseDerived &elseTensor) const
Definition: TensorBase.h:635
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_gamma_sample_der_alpha_op< Scalar >, const Derived, const OtherDerived > gamma_sample_der_alpha(const OtherDerived &other) const
Definition: TensorBase.h:230
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_cmp_op< Scalar, Scalar, internal::cmp_GE >, const Derived, const TensorCwiseNullaryOp< internal::scalar_constant_op< Scalar >, const Derived > > operator>=(Scalar threshold) const
Definition: TensorBase.h:601
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_y1_op< Scalar >, const Derived > bessel_y1() const
Definition: TensorBase.h:185
TensorScanOp< internal::ProdReducer< CoeffReturnType >, const Derived > TensorScanProdOp
Definition: TensorBase.h:676
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_max_op< Scalar, Scalar, NanPropagation >, const Derived, const TensorCwiseNullaryOp< internal::scalar_constant_op< Scalar >, const Derived > > cwiseMax(Scalar threshold) const
Definition: TensorBase.h:418
DerivedTraits::Index Index
Definition: TensorBase.h:37
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_cmp_op< Scalar, Scalar, internal::cmp_NEQ >, const Derived, const TensorCwiseNullaryOp< internal::scalar_constant_op< Scalar >, const Derived > > operator!=(Scalar threshold) const
Definition: TensorBase.h:611
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorInflationOp< const Strides, const Derived > inflate(const Strides &strides) const
Definition: TensorBase.h:976
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_real_op< Scalar >, const Derived > real() const
Definition: TensorBase.h:345
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorScanOp< Reducer, const Derived > scan(const Index &axis, const Reducer &reducer, bool exclusive=false) const
Definition: TensorBase.h:686
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorGeneratorOp< Generator, const Derived > generate(const Generator &generator) const
Definition: TensorBase.h:69
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_bitwise_or_op< Scalar >, const Derived, const OtherDerived > operator|(const OtherDerived &other) const
Definition: TensorBase.h:527
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_clamp_op< Scalar >, const Derived > clip(Scalar min, Scalar max) const
Definition: TensorBase.h:323
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_igamma_op< Scalar >, const Derived, const OtherDerived > igamma(const OtherDerived &other) const
Definition: TensorBase.h:216
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorFFTOp< const FFT, const Derived, FFTDataType, FFTDirection > fft(const FFT &dims) const
Definition: TensorBase.h:664
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorConcatenationOp< Axis, const Derived, const OtherDerived > concatenate(const OtherDerived &other, Axis axis) const
Definition: TensorBase.h:857
const TensorReductionOp< internal::ProdReducer< CoeffReturnType >, const DimensionList< Index, NumDimensions >, const Derived > prod() const
Definition: TensorBase.h:722
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_square_op< Scalar >, const Derived > square() const
Definition: TensorBase.h:107
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_cmp_op< Scalar, Scalar, internal::cmp_GE >, const Derived, const OtherDerived > operator>=(const TensorBase< OtherDerived, ReadOnlyAccessors > &other) const
Definition: TensorBase.h:567
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorPatchOp< const PatchDims, const Derived > extract_patches(const PatchDims &patch_dims) const
Definition: TensorBase.h:863
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_cmp_op< Scalar, Scalar, internal::cmp_NEQ >, const Derived, const OtherDerived > operator!=(const TensorBase< OtherDerived, ReadOnlyAccessors > &other) const
Definition: TensorBase.h:579
TensorScanOp< internal::SumReducer< CoeffReturnType >, const Derived > TensorScanSumOp
Definition: TensorBase.h:669
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_lgamma_op< Scalar >, const Derived > lgamma() const
Definition: TensorBase.h:131
const TensorReductionOp< internal::MinReducer< CoeffReturnType, NanPropagation >, const DimensionList< Index, NumDimensions >, const Derived > minimum() const
Definition: TensorBase.h:748
DerivedTraits::Scalar Scalar
Definition: TensorBase.h:36
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorShufflingOp< const Shuffle, const Derived > shuffle(const Shuffle &shfl) const
Definition: TensorBase.h:966
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_boolean_or_op< Scalar >, const Derived, const OtherDerived > operator||(const OtherDerived &other) const
Definition: TensorBase.h:515
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const std::enable_if_t< internal::is_arithmetic< typename NumTraits< ScalarExponent >::Real >::value, TensorCwiseUnaryOp< internal::scalar_unary_pow_op< Scalar, ScalarExponent >, const Derived > > pow(ScalarExponent exponent) const
Definition: TensorBase.h:338
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_i0e_op< Scalar >, const Derived > bessel_i0e() const
Definition: TensorBase.h:149
const TensorWithFormat< Derived, DerivedTraits::Layout, DerivedTraits::NumDimensions, Format > format(const Format &fmt) const
Definition: TensorBase.h:1007
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_inverse_op< Scalar >, const Derived > inverse() const
Definition: TensorBase.h:119
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_log_op< Scalar >, const Derived > log() const
Definition: TensorBase.h:293
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_igammac_op< Scalar >, const Derived, const OtherDerived > igammac(const OtherDerived &other) const
Definition: TensorBase.h:237
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorConvolutionOp< const Dimensions, const Derived, const KernelDerived > convolve(const KernelDerived &kernel, const Dimensions &dims) const
Definition: TensorBase.h:657
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_boolean_not_op< Scalar >, const Derived > operator!() const
Definition: TensorBase.h:539
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorPairReducerOp< internal::ArgMaxPairReducer< Pair< Index, CoeffReturnType > >, const array< Index, NumDimensions >, const Derived > argmax() const
Definition: TensorBase.h:783
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< CustomUnaryOp, const Derived > unaryExpr(const CustomUnaryOp &func) const
Definition: TensorBase.h:76
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp< internal::SumReducer< CoeffReturnType >, const Dims, const Derived > sum(const Dims &dims) const
Definition: TensorBase.h:693
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_y0_op< Scalar >, const Derived > bessel_y0() const
Definition: TensorBase.h:173
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_bitwise_xor_op< Scalar >, const Derived, const OtherDerived > operator^(const OtherDerived &other) const
Definition: TensorBase.h:533
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_exp_op< Scalar >, const Derived > exp() const
Definition: TensorBase.h:281
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_erf_op< Scalar >, const Derived > erf() const
Definition: TensorBase.h:257
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const std::conditional_t< internal::is_same< NewType, CoeffReturnType >::value, Derived, TensorConversionOp< NewType, const Derived > > cast() const
Definition: TensorBase.h:434
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReshapingOp< const NewDimensions, const Derived > reshape(const NewDimensions &newDimensions) const
Definition: TensorBase.h:919
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_cube_op< Scalar >, const Derived > cube() const
Definition: TensorBase.h:113
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_rsqrt_op< Scalar >, const Derived > rsqrt() const
Definition: TensorBase.h:101
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_cmp_op< Scalar, Scalar, internal::cmp_GT >, const Derived, const OtherDerived > operator>(const TensorBase< OtherDerived, ReadOnlyAccessors > &other) const
Definition: TensorBase.h:562
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorImagePatchOp< Dynamic, Dynamic, const Derived > extract_image_patches(const Index patch_rows, const Index patch_cols, const Index row_stride, const Index col_stride, const Index in_row_stride, const Index in_col_stride, const Index row_inflate_stride, const Index col_inflate_stride, const Index padding_top, const Index padding_bottom, const Index padding_left, const Index padding_right, const Scalar padding_value) const
Definition: TensorBase.h:879
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bitwise_not_op< Scalar >, const Derived > operator~() const
Definition: TensorBase.h:545
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_k0_op< Scalar >, const Derived > bessel_k0() const
Definition: TensorBase.h:191
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_product_op< Scalar >, const Derived, const OtherDerived > operator*(const OtherDerived &other) const
Definition: TensorBase.h:484
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorPaddingOp< const PaddingDimensions, const Derived > pad(const PaddingDimensions &padding, const Scalar padding_value) const
Definition: TensorBase.h:961
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_i0_op< Scalar >, const Derived > bessel_i0() const
Definition: TensorBase.h:143
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< CustomBinaryOp, const Derived, const OtherDerived > binaryExpr(const OtherDerived &other, const CustomBinaryOp &func) const
Definition: TensorBase.h:465
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_i1_op< Scalar >, const Derived > bessel_i1() const
Definition: TensorBase.h:155
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_bitwise_and_op< Scalar >, const Derived, const OtherDerived > operator&(const OtherDerived &other) const
Definition: TensorBase.h:521
const TensorTraceOp< const DimensionList< Index, NumDimensions >, const Derived > trace() const
Definition: TensorBase.h:844
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_digamma_op< Scalar >, const Derived > digamma() const
Definition: TensorBase.h:137
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorSlicingOp< const StartIndices, const Sizes, const Derived > slice(const StartIndices &startIndices, const Sizes &sizes) const
Definition: TensorBase.h:924
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorPaddingOp< const PaddingDimensions, const Derived > pad(const PaddingDimensions &padding) const
Definition: TensorBase.h:956
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReverseOp< const ReverseDimensions, const Derived > reverse(const ReverseDimensions &rev) const
Definition: TensorBase.h:946
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorBroadcastingOp< const Broadcast, const Derived > broadcast(const Broadcast &bcast) const
Definition: TensorBase.h:851
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_igamma_der_a_op< Scalar >, const Derived, const OtherDerived > igamma_der_a(const OtherDerived &other) const
Definition: TensorBase.h:223
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorPairReducerOp< internal::ArgMinPairReducer< Pair< Index, CoeffReturnType > >, const array< Index, 1 >, const Derived > argmin(const Index return_dim) const
Definition: TensorBase.h:822
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_expm1_op< Scalar >, const Derived > expm1() const
Definition: TensorBase.h:287
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorContractionOp< const Dimensions, const Derived, const OtherDerived, const OutputKernel > contract(const OtherDerived &other, const Dimensions &dims, const OutputKernel &output_kernel) const
Definition: TensorBase.h:650
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_quotient_op< Scalar >, const Derived, const OtherDerived > operator/(const OtherDerived &other) const
Definition: TensorBase.h:490
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_difference_op< Scalar >, const Derived, const OtherDerived > operator-(const OtherDerived &other) const
Definition: TensorBase.h:478
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorPairReducerOp< internal::ArgMaxPairReducer< Pair< Index, CoeffReturnType > >, const array< Index, 1 >, const Derived > argmax(const Index return_dim) const
Definition: TensorBase.h:809
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_log1p_op< Scalar >, const Derived > log1p() const
Definition: TensorBase.h:299
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorContractionOp< const Dimensions, const Derived, const OtherDerived, const NoOpOutputKernel > contract(const OtherDerived &other, const Dimensions &dims) const
Definition: TensorBase.h:644
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorStridingSlicingOp< const StartIndices, const StopIndices, const Strides, const Derived > stridedSlice(const StartIndices &startIndices, const StopIndices &stopIndices, const Strides &strides) const
Definition: TensorBase.h:929
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_logistic_op< Scalar >, const Derived > sigmoid() const
Definition: TensorBase.h:275
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_rint_op< Scalar >, const Derived > rint() const
Definition: TensorBase.h:446
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_cmp_op< Scalar, Scalar, internal::cmp_GT >, const Derived, const TensorCwiseNullaryOp< internal::scalar_constant_op< Scalar >, const Derived > > operator>(Scalar threshold) const
Definition: TensorBase.h:596
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp< internal::OrReducer, const DimensionList< Index, NumDimensions >, const std::conditional_t< internal::is_same< bool, CoeffReturnType >::value, Derived, TensorConversionOp< bool, const Derived > > > any() const
Definition: TensorBase.h:774
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorStridingOp< const Strides, const Derived > stride(const Strides &strides) const
Definition: TensorBase.h:971
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorForcedEvalOp< const Derived > eval() const
Definition: TensorBase.h:1001
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_k1e_op< Scalar >, const Derived > bessel_k1e() const
Definition: TensorBase.h:209
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_erfc_op< Scalar >, const Derived > erfc() const
Definition: TensorBase.h:263
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_floor_op< Scalar >, const Derived > floor() const
Definition: TensorBase.h:458
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const std::conditional_t< NumTraits< CoeffReturnType >::IsComplex, TensorCwiseUnaryOp< internal::scalar_conjugate_op< Scalar >, const Derived >, Derived > conjugate() const
Definition: TensorBase.h:331
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp< internal::OrReducer, const Dims, const std::conditional_t< internal::is_same< bool, CoeffReturnType >::value, Derived, TensorConversionOp< bool, const Derived > > > any(const Dims &dims) const
Definition: TensorBase.h:768
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_cmp_op< Scalar, Scalar, internal::cmp_EQ >, const Derived, const OtherDerived > operator==(const TensorBase< OtherDerived, ReadOnlyAccessors > &other) const
Definition: TensorBase.h:573
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_k0e_op< Scalar >, const Derived > bessel_k0e() const
Definition: TensorBase.h:197
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCustomUnaryOp< const CustomUnaryFunc, const Derived > customOp(const CustomUnaryFunc &op) const
Definition: TensorBase.h:990
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp< internal::ProdReducer< CoeffReturnType >, const Dims, const Derived > prod(const Dims &dims) const
Definition: TensorBase.h:717
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_i1e_op< Scalar >, const Derived > bessel_i1e() const
Definition: TensorBase.h:161
const TensorReductionOp< internal::MaxReducer< CoeffReturnType, NanPropagation >, const DimensionList< Index, NumDimensions >, const Derived > maximum() const
Definition: TensorBase.h:735
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorImagePatchOp< Dynamic, Dynamic, const Derived > extract_image_patches(const Index patch_rows=1, const Index patch_cols=1, const Index row_stride=1, const Index col_stride=1, const Index in_row_stride=1, const Index in_col_stride=1, const PaddingType padding_type=PADDING_SAME, const Scalar padding_value=Scalar(0)) const
Definition: TensorBase.h:869
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorVolumePatchOp< Dynamic, Dynamic, Dynamic, const Derived > extract_volume_patches(const Index patch_planes, const Index patch_rows, const Index patch_cols, const Index plane_stride=1, const Index row_stride=1, const Index col_stride=1, const PaddingType padding_type=PADDING_SAME, const Scalar padding_value=Scalar(0)) const
Definition: TensorBase.h:893
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_min_op< Scalar, Scalar, NaNPropagation >, const Derived, const OtherDerived > cwiseMin(const OtherDerived &other) const
Definition: TensorBase.h:502
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_sum_op< Scalar >, const Derived, const OtherDerived > operator+(const OtherDerived &other) const
Definition: TensorBase.h:472
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_k1_op< Scalar >, const Derived > bessel_k1() const
Definition: TensorBase.h:203
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseNullaryOp< internal::scalar_constant_op< Scalar >, const Derived > constant(const Scalar &value) const
Definition: TensorBase.h:51
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorChippingOp< Dynamic, const Derived > chip(const Index offset, const Index dim) const
Definition: TensorBase.h:941
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_tanh_op< Scalar >, const Derived > tanh() const
Definition: TensorBase.h:125
const TensorReductionOp< internal::MeanReducer< CoeffReturnType >, const DimensionList< Index, NumDimensions >, const Derived > mean() const
Definition: TensorBase.h:710
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Derived & derived() const
Definition: TensorBase.h:1016
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReductionOp< internal::AndReducer, const DimensionList< Index, NumDimensions >, const std::conditional_t< internal::is_same< bool, CoeffReturnType >::value, Derived, TensorConversionOp< bool, const Derived > > > all() const
Definition: TensorBase.h:761
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseNullaryOp< RandomGenerator, const Derived > random(const RandomGenerator &gen=RandomGenerator()) const
Definition: TensorBase.h:62
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorScanSumOp cumsum(const Index &axis, bool exclusive=false) const
Definition: TensorBase.h:672
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_boolean_and_op< Scalar >, const Derived, const OtherDerived > operator&&(const OtherDerived &other) const
Definition: TensorBase.h:509
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorVolumePatchOp< Dynamic, Dynamic, Dynamic, const Derived > extract_volume_patches(const Index patch_planes, const Index patch_rows, const Index patch_cols, const Index plane_stride, const Index row_stride, const Index col_stride, const Index plane_inflate_stride, const Index row_inflate_stride, const Index col_inflate_stride, const Index padding_top_z, const Index padding_bottom_z, const Index padding_top, const Index padding_bottom, const Index padding_left, const Index padding_right, const Scalar padding_value=Scalar(0)) const
Definition: TensorBase.h:902
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorIndexPairOp< const Derived > index_pairs() const
Definition: TensorBase.h:983
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCustomBinaryOp< const CustomBinaryFunc, const Derived, const OtherDerived > customOp(const OtherDerived &other, const CustomBinaryFunc &op) const
Definition: TensorBase.h:995
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseBinaryOp< internal::scalar_min_op< Scalar, Scalar, NanPropagation >, const Derived, const TensorCwiseNullaryOp< internal::scalar_constant_op< Scalar >, const Derived > > cwiseMin(Scalar threshold) const
Definition: TensorBase.h:425
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_bessel_j1_op< Scalar >, const Derived > bessel_j1() const
Definition: TensorBase.h:179
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_sign_op< Scalar >, const Derived > sign() const
Definition: TensorBase.h:95
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorRollOp< const Rolls, const Derived > roll(const Rolls &rolls) const
Definition: TensorBase.h:951
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorCwiseUnaryOp< internal::scalar_ceil_op< Scalar >, const Derived > ceil() const
Definition: TensorBase.h:452
The tensor base class.
Definition: TensorBase.h:1026
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorChippingOp< DimId, Derived > chip(const Index offset)
Definition: TensorBase.h:1147
TensorBase< Derived, ReadOnlyAccessors > Base
Definition: TensorBase.h:1028
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & setRandom()
Definition: TensorBase.h:1049
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorLayoutSwapOp< const Derived > swap_layout() const
Definition: TensorBase.h:1084
DerivedTraits::Scalar Scalar
Definition: TensorBase.h:1030
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReshapingOp< const NewDimensions, const Derived > reshape(const NewDimensions &newDimensions) const
Definition: TensorBase.h:1106
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorStridingOp< const Strides, Derived > stride(const Strides &strides)
Definition: TensorBase.h:1203
TensorDevice< Derived, DeviceType > device(const DeviceType &dev)
Definition: TensorBase.h:1209
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorShufflingOp< const Shuffle, Derived > shuffle(const Shuffle &shfl)
Definition: TensorBase.h:1192
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorStridingSlicingOp< const StartIndices, const StopIndices, const Strides, const Derived > stridedSlice(const StartIndices &startIndices, const StopIndices &stopIndices, const Strides &strides) const
Definition: TensorBase.h:1128
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorConcatenationOp< const Axis, Derived, OtherDerived > concatenate(const OtherDerived &other, const Axis &axis)
Definition: TensorBase.h:1100
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorReverseOp< const ReverseDimensions, const Derived > reverse(const ReverseDimensions &rev) const
Definition: TensorBase.h:1165
static constexpr int NumDimensions
Definition: TensorBase.h:1033
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorConcatenationOp< const Axis, const Derived, const OtherDerived > concatenate(const OtherDerived &other, const Axis &axis) const
Definition: TensorBase.h:1095
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorChippingOp< Dynamic, Derived > chip(const Index offset, const Index dim)
Definition: TensorBase.h:1159
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & setValues(const typename internal::Initializer< Derived, NumDimensions >::InitList &vals)
Definition: TensorBase.h:1058
internal::traits< Derived > DerivedTraits
Definition: TensorBase.h:1029
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorChippingOp< Dynamic, const Derived > chip(const Index offset, const Index dim) const
Definition: TensorBase.h:1154
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & operator*=(const OtherDerived &other)
Definition: TensorBase.h:1074
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorSlicingOp< const StartIndices, const Sizes, const Derived > slice(const StartIndices &startIndices, const Sizes &sizes) const
Definition: TensorBase.h:1117
Scalar CoeffReturnType
Definition: TensorBase.h:1032
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & setConstant(const Scalar &val)
Definition: TensorBase.h:1045
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorLayoutSwapOp< Derived > swap_layout()
Definition: TensorBase.h:1089
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & setRandom()
Definition: TensorBase.h:1053
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorSlicingOp< const StartIndices, const Sizes, Derived > slice(const StartIndices &startIndices, const Sizes &sizes)
Definition: TensorBase.h:1122
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & setZero()
Definition: TensorBase.h:1041
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & derived()
Definition: TensorBase.h:1220
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & operator-=(const OtherDerived &other)
Definition: TensorBase.h:1070
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorRollOp< const Rolls, Derived > roll(const Rolls &roll)
Definition: TensorBase.h:1181
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Derived & derived() const
Definition: TensorBase.h:1222
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorStridingSlicingOp< const StartIndices, const StopIndices, const Strides, Derived > stridedSlice(const StartIndices &startIndices, const StopIndices &stopIndices, const Strides &strides)
Definition: TensorBase.h:1134
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorRollOp< const Rolls, const Derived > roll(const Rolls &roll) const
Definition: TensorBase.h:1176
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorStridingOp< const Strides, const Derived > stride(const Strides &strides) const
Definition: TensorBase.h:1198
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorChippingOp< DimId, const Derived > chip(const Index offset) const
Definition: TensorBase.h:1141
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & operator+=(const OtherDerived &other)
Definition: TensorBase.h:1066
TensorAsyncDevice< Derived, DeviceType, DoneCallback > device(const DeviceType &dev, DoneCallback done)
Definition: TensorBase.h:1215
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & operator/=(const OtherDerived &other)
Definition: TensorBase.h:1078
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const TensorShufflingOp< const Shuffle, const Derived > shuffle(const Shuffle &shfl) const
Definition: TensorBase.h:1187
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorReverseOp< const ReverseDimensions, Derived > reverse(const ReverseDimensions &rev)
Definition: TensorBase.h:1170
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Derived & operator=(const OtherDerived &other)
Definition: TensorBase.h:1233
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorReshapingOp< const NewDimensions, Derived > reshape(const NewDimensions &newDimensions)
Definition: TensorBase.h:1111
DerivedTraits::Index Index
Definition: TensorBase.h:1031
Definition: TensorBroadcasting.h:66
Definition: TensorChipping.h:70
Tensor concatenation class.
Definition: TensorConcatenation.h:60
Definition: TensorContraction.h:307
Tensor conversion class. This class makes it possible to vectorize type casting operations when the n...
Definition: TensorConversion.h:166
Definition: TensorConvolution.h:236
Tensor custom class.
Definition: TensorCustomOp.h:195
Tensor custom class.
Definition: TensorCustomOp.h:51
Definition: TensorExpr.h:162
Definition: TensorExpr.h:49
Definition: TensorExpr.h:97
Pseudo expression providing an operator = that will evaluate its argument on the specified computing ...
Definition: TensorDevice.h:31
Definition: TensorFFT.h:109
The fixed sized version of the tensor class.
Definition: TensorFixedSize.h:30
Definition: TensorForcedEval.h:57
Tensor generator class.
Definition: TensorGenerator.h:52
Definition: TensorImagePatch.h:117
Definition: TensorArgMax.h:52
Definition: TensorInflation.h:52
Definition: TensorLayoutSwap.h:67
Definition: TensorPadding.h:53
Definition: TensorArgMax.h:154
Definition: TensorPatch.h:52
Definition: TensorReduction.h:510
Definition: TensorMorphing.h:53
Definition: TensorReverse.h:52
Definition: TensorRoll.h:50
Definition: TensorScan.h:49
Definition: TensorExpr.h:302
Definition: TensorShuffling.h:52
Definition: TensorMorphing.h:289
Definition: TensorStriding.h:52
Definition: TensorMorphing.h:752
Definition: TensorTrace.h:53
Definition: TensorVolumePatch.h:56
The tensor class.
Definition: Tensor.h:68
static EIGEN_STRONG_INLINE void run(const Expression &expr, const Device &device=DefaultDevice())
Definition: TensorExecutor.h:92
@ IsComplex
Definition: common.h:73
std::vector< Array2i > sizes
Definition: dense_solvers.cpp:12
@ ReadOnlyAccessors
Definition: Constants.h:372
char char * op
Definition: level2_impl.h:374
#define isfinite(X)
Definition: main.h:111
#define isnan(X)
Definition: main.h:109
#define isinf(X)
Definition: main.h:110
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC bfloat16 operator/(const bfloat16 &a, const bfloat16 &b)
Definition: BFloat16.h:284
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC bool operator>=(const bfloat16 &a, const bfloat16 &b)
Definition: BFloat16.h:340
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC bool operator>(const bfloat16 &a, const bfloat16 &b)
Definition: BFloat16.h:337
EIGEN_ALWAYS_INLINE DSizes< IndexType, NumDims > strides(const DSizes< IndexType, NumDims > &dimensions)
Definition: TensorBlock.h:29
Namespace containing all symbols from the Eigen library.
Definition: bench_norm.cpp:70
std::array< T, N > array
Definition: EmulateArray.h:231
squared absolute value
Definition: GlobalFunctions.h:87
EIGEN_STRONG_INLINE const CwiseBinaryOp< internal::scalar_difference_op< typename DenseDerived::Scalar, typename SparseDerived::Scalar >, const DenseDerived, const SparseDerived > operator-(const MatrixBase< DenseDerived > &a, const SparseMatrixBase< SparseDerived > &b)
Definition: SparseCwiseBinaryOp.h:922
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE const T1 & choose(Cond< true >, const T1 &first, const T2 &)
Definition: TensorMeta.h:22
PaddingType
Definition: TensorTraits.h:227
@ PADDING_SAME
Definition: TensorTraits.h:227
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index
The Index type as used for the API.
Definition: Meta.h:83
CleanedUpDerType< DerType >::type() min(const AutoDiffScalar< DerType > &x, const T &y)
Definition: AutoDiffScalar.h:494
CleanedUpDerType< DerType >::type() max(const AutoDiffScalar< DerType > &x, const T &y)
Definition: AutoDiffScalar.h:499
EIGEN_DEVICE_FUNC const Product< MatrixDerived, PermutationDerived, AliasFreeProduct > operator*(const MatrixBase< MatrixDerived > &matrix, const PermutationBase< PermutationDerived > &permutation)
Definition: PermutationMatrix.h:471
EIGEN_CONSTEXPR EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool operator!=(const Pair< U, V > &x, const Pair< U, V > &y)
Definition: TensorMeta.h:232
EIGEN_CONSTEXPR EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool operator==(const Pair< U, V > &x, const Pair< U, V > &y)
Definition: TensorMeta.h:227
EIGEN_STRONG_INLINE const CwiseBinaryOp< internal::scalar_sum_op< typename DenseDerived::Scalar, typename SparseDerived::Scalar >, const DenseDerived, const SparseDerived > operator+(const MatrixBase< DenseDerived > &a, const SparseMatrixBase< SparseDerived > &b)
Definition: SparseCwiseBinaryOp.h:904
val
Definition: calibrate.py:119
internal::nested_eval< T, 1 >::type eval(const T &xpr)
Definition: sparse_permutations.cpp:47
Definition: TensorMeta.h:19
Definition: TensorDeviceDefault.h:19
Definition: TensorDimensionList.h:29
Definition: TensorMeta.h:238
Holds information about the various numeric (i.e. scalar) types allowed by Eigen.
Definition: NumTraits.h:217
Definition: TensorDimensions.h:85
A cost model used to limit the number of threads used for evaluating tensor expression.
Definition: TensorEvaluator.h:31
Definition: TensorFunctors.h:296
Definition: TensorFunctors.h:321
Definition: TensorFunctors.h:341
std::initializer_list< typename Initializer< Derived, N - 1 >::InitList > InitList
Definition: TensorInitializer.h:29
Definition: TensorFunctors.h:177
Definition: TensorFunctors.h:107
Definition: TensorFunctors.h:217
Definition: TensorFunctors.h:307
Definition: TensorFunctors.h:257
Definition: TensorFunctors.h:66
Definition: BinaryFunctors.h:716
Definition: BinaryFunctors.h:738
Template functor to compute the absolute value of a scalar.
Definition: functors/UnaryFunctors.h:44
Template functor to compute the phase angle of a complex.
Definition: functors/UnaryFunctors.h:161
Template functor to compute the modified Bessel function of the first kind of order zero.
Definition: BesselFunctionsFunctors.h:27
Template functor to compute the exponentially scaled modified Bessel function of the first kind of or...
Definition: BesselFunctionsFunctors.h:52
Template functor to compute the modified Bessel function of the first kind of order one.
Definition: BesselFunctionsFunctors.h:76
Template functor to compute the exponentially scaled modified Bessel function of the first kind of or...
Definition: BesselFunctionsFunctors.h:101
Template functor to compute the Bessel function of the second kind of order zero.
Definition: BesselFunctionsFunctors.h:125
Template functor to compute the Bessel function of the first kind of order one.
Definition: BesselFunctionsFunctors.h:175
Template functor to compute the modified Bessel function of the second kind of order zero.
Definition: BesselFunctionsFunctors.h:225
Template functor to compute the exponentially scaled modified Bessel function of the second kind of o...
Definition: BesselFunctionsFunctors.h:250
Template functor to compute the modified Bessel function of the second kind of order one.
Definition: BesselFunctionsFunctors.h:275
Template functor to compute the exponentially scaled modified Bessel function of the second kind of o...
Definition: BesselFunctionsFunctors.h:300
Template functor to compute the Bessel function of the second kind of order zero.
Definition: BesselFunctionsFunctors.h:150
Template functor to compute the Bessel function of the second kind of order one.
Definition: BesselFunctionsFunctors.h:200
Template functor to compute the bitwise and of two scalars.
Definition: BinaryFunctors.h:588
Template functor to compute the bitwise not of a scalar.
Definition: functors/UnaryFunctors.h:1145
Template functor to compute the bitwise or of two scalars.
Definition: BinaryFunctors.h:612
Template functor to compute the bitwise xor of two scalars.
Definition: BinaryFunctors.h:636
Template functor to compute the and of two scalars as if they were booleans.
Definition: BinaryFunctors.h:461
Template functor to compute the logical not of a scalar as if it were a boolean.
Definition: functors/UnaryFunctors.h:1099
Template functor to compute the or of two scalars as if they were booleans.
Definition: BinaryFunctors.h:489
Template functor to cast a scalar to another type.
Definition: functors/UnaryFunctors.h:205
Template functor to compute the ceil of a scalar.
Definition: functors/UnaryFunctors.h:954
Definition: TensorFunctors.h:399
Template functors for comparison of two scalars.
Definition: BinaryFunctors.h:199
Template functor to compute the conjugate of a complex value.
Definition: functors/UnaryFunctors.h:132
Definition: NullaryFunctors.h:21
Template functor to compute the cube of a scalar.
Definition: functors/UnaryFunctors.h:863
Template functor to compute the difference of two scalars.
Definition: BinaryFunctors.h:381
Template functor to compute psi, the derivative of lgamma of a scalar.
Definition: SpecialFunctionsFunctors.h:177
Template functor to compute the error function of a scalar.
Definition: SpecialFunctionsFunctors.h:247
Template functor to compute the Complementary Error Function of a scalar.
Definition: SpecialFunctionsFunctors.h:280
Template functor to compute the exponential of a scalar.
Definition: functors/UnaryFunctors.h:348
Template functor to compute the exponential of a scalar - 1.
Definition: functors/UnaryFunctors.h:402
Template functor to compute the floor of a scalar.
Definition: functors/UnaryFunctors.h:914
Template functor to compute the derivative of the sample of a Gamma(alpha, 1) random variable with re...
Definition: SpecialFunctionsFunctors.h:80
Template functor to compute the derivative of the incomplete gamma function igamma_der_a(a,...
Definition: SpecialFunctionsFunctors.h:53
Template functor to compute the incomplete gamma function igamma(a, x)
Definition: SpecialFunctionsFunctors.h:27
Template functor to compute the complementary incomplete gamma function igammac(a,...
Definition: SpecialFunctionsFunctors.h:105
Template functor to extract the imaginary part of a complex.
Definition: functors/UnaryFunctors.h:296
Template functor to compute the inverse of a scalar.
Definition: functors/UnaryFunctors.h:808
Definition: functors/UnaryFunctors.h:1074
Definition: functors/UnaryFunctors.h:1040
Definition: functors/UnaryFunctors.h:1005
Template functor to compute the natural log of the absolute value of Gamma of a scalar.
Definition: SpecialFunctionsFunctors.h:155
Template functor to compute the logarithm of 1 plus a scalar value.
Definition: functors/UnaryFunctors.h:456
Template functor to compute the base-2 logarithm of a scalar.
Definition: functors/UnaryFunctors.h:497
Template functor to compute the logarithm of a scalar.
Definition: functors/UnaryFunctors.h:424
Template functor to compute the logistic function of a scalar.
Definition: functors/UnaryFunctors.h:1215
Template functor to compute the max of two scalars.
Definition: BinaryFunctors.h:171
Template functor to compute the min of two scalars.
Definition: BinaryFunctors.h:142
Template functor to compute the modulo between an array and a scalar.
Definition: TensorFunctors.h:23
Template functor to compute the Inverse of the normal distribution function of a scalar.
Definition: SpecialFunctionsFunctors.h:303
Template functor to compute the opposite of a scalar.
Definition: functors/UnaryFunctors.h:26
Template functor to compute the polygamma function.
Definition: SpecialFunctionsFunctors.h:223
Template functor to compute the product of two scalars.
Definition: BinaryFunctors.h:73
Template functor to compute the quotient of two scalars.
Definition: BinaryFunctors.h:430
Template functor to extract the real part of a complex.
Definition: functors/UnaryFunctors.h:281
Template functor to compute the rounded (with current rounding mode) value of a scalar.
Definition: functors/UnaryFunctors.h:934
Template functor to compute the rounded value of a scalar.
Definition: functors/UnaryFunctors.h:894
Template functor to compute the reciprocal square root of a scalar.
Definition: functors/UnaryFunctors.h:573
Template functor to compute the signum of a scalar.
Definition: functors/UnaryFunctors.h:1168
Template functor to compute the square root of a scalar.
Definition: functors/UnaryFunctors.h:517
Template functor to compute the square of a scalar.
Definition: functors/UnaryFunctors.h:832
Template functor to compute the sum of two scalars.
Definition: BinaryFunctors.h:34
Template functor to compute the tanh of a scalar.
Definition: functors/UnaryFunctors.h:693
Definition: functors/UnaryFunctors.h:1325
Template functor to compute the Riemann Zeta function of two arguments.
Definition: SpecialFunctionsFunctors.h:199
Definition: benchGeometry.cpp:21