Tag: Argument

You can set keepdim with the functions which have keepdim argument as shown below: *Memos: 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) # tensor(10) torch.sum(input=my_tensor, dim=0, keepdim=True) # tensor([10]) prod(). *My...