Add prod op - Githubissues

csarofeen / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

http://pytorch.org

Other

26 stars 7 forks source link

Closed jacobhinkle closed 1 year ago

jacobhinkle commented 1 year ago

This adds the prod op which performs a product of elements in a tensor along a given set of dimensions.