torch.tensor转numpy

 

outputs.cpu().detach().numpy()[0,:,:,:].shape

import matplotlib.pyplot as pyplot 
pyplot.imshow(binary_mask)
pyplot.show()