SelectScalarFromRows.hpp File Reference

Select values from rows of a tensor. More...

#include <poplar/Graph.hpp>
#include <poplar/Program.hpp>

Go to the source code of this file.

Namespaces
namespace	popops
	Common functions, such as elementwise and reductions.

Functions
poplar::Tensor	popops::selectScalarFromRows (poplar::Graph &graph, const poplar::Tensor &params, const poplar::Tensor &indices, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
	For each row in the 2D tensor params, select a single scalar value. More...

Detailed Description

Select values from rows of a tensor.