Broadcast.hpp File Reference

Functions to provide numpy-like tensor matching and broadcasting. More...

#include <poplar/Tensor.hpp>

Go to the source code of this file.

Namespaces
namespace	poputil
	General utility functions for building graphs.

Functions
void	poputil::expandToMatchRanks (poplar::Tensor &a, poplar::Tensor &b)
	Match dimensions of two tensors using numpy-style expansion rules. More...
void	poputil::broadcastToMatch (poplar::Tensor &a, const std::vector< std::size_t > &shape)
	Match dimensions of a tensor to a shape using numpy-style broadcast rules: More...
void	poputil::broadcastToMatch (poplar::Tensor &a, poplar::Tensor &b)
	Match dimensions of two tensors using numpy-style broadcast rules: More...
void	poputil::broadcastToMatch (poplar::Tensor &a, poplar::Tensor &b, poplar::Tensor &c)
	Match dimensions of three tensors using numpy-style broadcast rules: More...
bool	poputil::canBroadcastToMatch (const poplar::Tensor &a, const poplar::Tensor &b)
	Test if the given tensors can be broadcast to match one another using the rules for broadcastToMatch(). More...

Detailed Description

Functions to provide numpy-like tensor matching and broadcasting.