I have a var
a with shape (time, grid_size, grid_size) and I have a torch tensor
b with shape (grid_size, grid_size). I would like to ask if I can use
And can I use the same on to_torch like b = a.to_torch()?
And can I use the same on fill like