Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

pytorch tensor argmax

values, indices = tensor.max(0)
values, indices = torch.max(tensor, 0)
Source by discuss.pytorch.org #
 
PREVIOUS NEXT
Tagged: #pytorch #tensor #argmax
ADD COMMENT
Topic
Name
8+7 =