Function tensor_to_img_array

func_2d/utils.py:284–287

(tensor)

def tensor_to_img_array(tensor):
    image = tensor.cpu().detach().numpy()
    image = np.transpose(image, [0, 2, 3, 1])
    return image


def view(tensor):