Собственная сумма (), чтобы плавать
У меня есть Eigen TensorMap в моей пользовательской операции тензорного потока типа
Eigen::TensorMap<Eigen::Tensor<float, 1, 1, long int>, 16, Eigen::MakePointer>
Подбор номера с него работает:
float a = Tbuf(0);
Но если хотите подвести его к скалярному числу
float b = Tbuf.sum();
Я получаю ошибку
word2vec_kernels.cc: In member function ‘virtual void tensorflow::NegTrainWord2vecOp::Compute(tensorflow::OpKernelContext*)’:
word2vec_kernels.cc:516:29: error: cannot convert ‘const Eigen::TensorReductionOp<Eigen::internal::SumReducer<float>, const Eigen::DimensionList<long int, 1ul>, const Eigen::TensorMap<Eigen::Tensor<float, 1, 1, long int>, 16, Eigen::MakePointer>, Eigen::MakePointer>’ to ‘float’ in initialization
float b = Tbuf.sum();
1 ответ
Решение
Хорошо, я забыл, что тензор должен быть оценен сначала.
float b = ((Eigen::Tensor<float, 0, 1, long int> )Tbuf.sum())(0);