Update TileOnGrid backend

[bhashemian]

Is your feature request related to a problem? Please describe.
Update TileOnGrid to support input/output type of torch.Tensor and numpy while its backend is numpy only.