Speed of torch.max

2

The answer to your question is the same reason as why someone thought .item() was the slowest operation: Tensor.item() takes a lot of running time

1 Like