텐서 요소가 경계를 벗어 났는지 여부를 확인할 수 있습니까?

Question

torch.cuda.FloatTensor을 으로 사용하여 텐서의 원소가 경계를 벗어 났는지 여부를 확인할 수 있습니까? PyTorch, GPU 접근 방법은?

예 (check limits) :

idxmax = (x > xmax) # elements that are bigger that upper limit 
idxmim = (x < min) # elements that are smaller that upper limit 
x[idxmax] = xmax 
x[idxmin] = xmin 

하지 않으면, 그것은 단지 CPU를 사용하여,이 check limits 부분을 할 수 있습니다 : 나는 다음을 시도했지만 속도를하지 않았다

for i in range(pop): 
    if (x[i]>xmax): 
     x[i]=xmax 
    elif (x[i]<xmin): 
     x[i]=xmin 

? 방법?

Answer 1

텐서 x의 CPU 복사본을 구한 다음 작업을 수행하고 텐서를 GPU 메모리로 다시 밀어 넣을 수 있습니다.

x = x.cpu() # get the CPU copy 
# do your operations 
x = x.cuda() # move the object back to cuda memory