Currently, all losses in pytorch returns a single number, which is either the sum of the losses per element, or the average.
Obtaining losses per element is not yet implemented in pytorch, and is being tracked in https://github.com/pytorch/pytorch/issues/264 508