latest/doxygen/MeshGrid_8hpp_source.html

// Copyright (c) 2018 Graphcore Ltd. All rights reserved.

#ifndef poplin_MeshGrid_hpp

#define poplin_MeshGrid_hpp


#include <poplar/Graph.hpp>


namespace poplin {


poplar::Tensor linspace(poplar::Graph &graph, const poplar::Type &type,

                        float left, float right, size_t count,

                        const poplar::DebugContext &debugContext = {});


std::vector<poplar::Tensor> meshgrid2d(poplar::Graph &graph, poplar::Tensor x,

                                       poplar::Tensor y);


} // end namespace poplin


#endif // poplin_MeshGrid_hpp

poplar::DebugContext
DebugContext gathers the common external parameters of the context of an operation.
Definition: DebugContext.hpp:221

poplar::Graph
This class represents a graph program to be executed on the IPU.
Definition: Graph.hpp:52

poplar::Tensor
A reference to a subset of tensor elements.
Definition: Tensor.hpp:38

poplar::Type
Class representing device data types.
Definition: Type.hpp:42

poplin
Linear algebra functions.
Definition: Cholesky.hpp:14

poplin::linspace
poplar::Tensor linspace(poplar::Graph &graph, const poplar::Type &type, float left, float right, size_t count, const poplar::DebugContext &debugContext={})
Create a constant variable that contains values equally spaced in the specified closed range [left,...

poplin::meshgrid2d
std::vector< poplar::Tensor > meshgrid2d(poplar::Graph &graph, poplar::Tensor x, poplar::Tensor y)
Create a coordinate grid for each axis by broadcasting the input tensors.