Set keepdim with keepdim argument functions PyTorch

You can set keepdim with the functions which have keepdim argument as shown below:

*Memos:

• I selected some popular keepdim argument functions such as sum(),
  prod()
  mean(),
  median(),
  min(), max(),
  argmin(),
  argmax(),
  all() and any().


• 
  keepdim(bool, optional-Default:False) can keep the dimension of the input tensor.


• Sometimes, keepdim needs to be used with dim.

sum(). *My post explains sum():

import torch

my_tensor = torch.tensor([1, 2, 3, 4])

torch.sum(input=my_tensor)
torch.sum(input=my_tensor, dim=0)
# …

argmin argument false function functions input keepdim max mean median memos min popular prod pytorch set sum tensor you