Hello,
Is it possible to use the GPU to get the min or max value of a tensor?
Example:
Find the min value of x
[pop x dim], pop = 1000000 and N = 1000.
Thank you.
Hello,
Is it possible to use the GPU to get the min or max value of a tensor?
Example:
Find the min value of x
[pop x dim], pop = 1000000 and N = 1000.
Thank you.
min = torch.min(x)
max = torch.max(x)