Take the mean over all non-batch dimensions.
(tensor)
source not stored for this graph (policy: none)
no outgoing calls
no test coverage detected