20f92bda2e4dd571e85d2bcb4c3ebb3448af7e69,kornia/augmentation/functional.py,,random_grayscale,#Any#Any#Any#,95

Before Change

    See :func:`~kornia.augmentation.functional.apply_grayscale` for details.
    

    if isinstance(input, tuple):
        batch_size = input[0].shape[0] if len(input[0].shape) == 4 else 1
    else:
        batch_size = input.shape[0] if len(input.shape) == 4 else 1
    params = rg.random_prob_generator(batch_size, p=p)

    output = apply_grayscale(input, params)
    if return_transform:

After Change

    See :func:`~kornia.augmentation.random_generator.random_prob_generator` for details.
    See :func:`~kornia.augmentation.functional.apply_grayscale` for details.
    
    input = _transform_input(input)
    batch_size, _, h, w = input.size()
    params = rg.random_prob_generator(batch_size, p=p)

    output = apply_grayscale(input, params)
    if return_transform:

Instances