- 在C语言中,我们要在函数中使用全局变量,只需要没有相应的同名同类型局部变量就可以做到,不需要有相应的关键字。
- 在python中,因为没有类型声明,因此当我们要使用全局变量的时候,需要global关键字来声明。
例如:
pixel_coords = None
def set_id_grid(depth):
global pixel_coords
b, h, w = depth.size()
i_range = torch.arange(0, h).view(1, h, 1).expand(1,h,w).type_as(depth) # [1, H, W]
j_range = torch.arange(0, w).view(1, 1, w).expand(1,h,w).type_as(depth) # [1, H, W]
ones = torch.ones(1,h,w).type_as(depth)
pixel_coords = torch.stack((j_range, i_range, ones), dim=1) # [1, 3, H, W]