during normalize, construct mean and std tensors on the same device as the input tensor (#787)
during normalize, construct mean and std tensors on the same device as the input tensor (#787)


pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision