У меня есть тензор t размерности [B, C, H, W].Я использовал view(), чтобы сделать его двухмерным, и поместил его в какой-то файл.
t
[B, C, H, W]
view()
tensor.view(B, -1)
Теперь во время чтения файла я хочу восстановить сброшенный тензор.Как восстановить это правильно?