SpatialSoftMax.hpp File Reference

Functions for spatial softmax. More...

#include <poplar/Graph.hpp>
#include <popnn/NonLinearity.hpp>

Go to the source code of this file.

Namespaces
namespace	popnn
	Functions used in neural networks.

Functions
std::pair< poplar::Tensor, poplar::Tensor >	popnn::spatialSoftMax2D (poplar::Graph &graph, poplar::program::Sequence &prog, const poplar::Tensor &fields, float temperature, bool disableSoftmax=false, const poplar::DebugContext &debugContext={})
	Implements a spatial softmax specialised for 2D input fields. More...

Detailed Description

Functions for spatial softmax.