Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

Repository URL to install this package:

Details    
torchvision / ops / new_empty_tensor.py
Size: Mime:
import torch
from torch.jit.annotations import List
from torch import Tensor


def _new_empty_tensor(x, shape):
    # type: (Tensor, 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)