Could not work in Pytorch about python-image-preview

[076923]

First of all, thank you for the question. I have updated extension. (0.0.9 update)

CUDA support and torch shape operations were improved.

You can now use it like this:

import torch

img1 = torch.full((500, 500), 127, dtype=torch.uint8)          # 1-Channel Gray Image
img2 = torch.zeros((1, 500, 500))                              # 1-Channel Black Image
img3 = torch.zeros((500, 500))                                 # 1-Channel Black Image
img4 = torch.full((3, 500, 500), 1.0, dtype=torch.float32)     # 3-Channel White Image
img5 = torch.full((500, 500), 0.5, device='cuda')              # 1-Channel Gray Imaeg
img6 = torch.rand_like(img5)                                   # 1-Channel Random Image