I am doing some operations on batch size 100 and tensors of size (784, 300), (300, 100), (100, 10). but mostly i work on the average overall the batches. so i use torch.mean to take the average of tensors on the batch and then use torch.bmm and torch.ger and torch.eig on tensors of size (784, 300), (300, 100), (100, 10).