code: def test_return_max_across_color_channels_if_specified(mocker, model):
      backprop = Backprop(model)
  
      target_class = 5
      input_ = torch.zeros([1, 3, 224, 224])
  
      make_mock_output(mocker, model, target_class)
  
      gradients = backprop.calculate_gradients(
          input_, target_class, take_max=True
      )
  
      assert gradients.shape == (1, 224, 224)
  
imports: import inspect
  import pytest
  import torch
  import torch.nn
  import torch.nn.functional
  import torchvision.models
  from flashtorch.saliency import Backprop
label_desc: Use image node_id 14 for calling a global function with example usage: test_return_max_across_color_channels_if_specified(mocker, model) without return types