infinityfuture/pytorch_global_max_pooling.py

Last active October 3, 2018 16:47

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/infinityfuture/ea37d68fd3476c55a0b7b92d747d9e7b.js"></script>
Save infinityfuture/ea37d68fd3476c55a0b7b92d747d9e7b to your computer and use it in GitHub Desktop.

Download ZIP

Pytorch Global Max Pooling

Raw

pytorch_global_max_pooling.py

	def global_max_pooling(tensor, dim, topk):
	"""Global max pooling"""
	ret, _ = torch.topk(tensor, topk, dim)
	return ret

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment