如果扩充一个维度可以考虑tf.expand_dims,但是扩充多个维度就显得吃力了。直接上代码:
grid_x = torch.tensor([-1, 0, 1, 0.5], dtype=torch.float)
grid_x = grid_x[None, None, :, None] # 扩充维度至[1,1,4,1]
如果扩充一个维度可以考虑tf.expand_dims,但是扩充多个维度就显得吃力了。直接上代码:
grid_x = torch.tensor([-1, 0, 1, 0.5], dtype=torch.float)
grid_x = grid_x[None, None, :, None] # 扩充维度至[1,1,4,1]