new_empty_tensor.py

import torch
from torch.jit.annotations import List
from torch import Tensor


def _new_empty_tensor(x: Tensor, shape: List[int]) -> Tensor:
    """
    Arguments:
        input (Tensor): input tensor
        shape List[int]: the new empty tensor shape

    Returns:
        output (Tensor)
    """
    return torch.ops.torchvision._new_empty_tensor_op(x, shape)