Function pb2torch

python_backend/model.py:13–16

(request, name)

def pb2torch(request, name):
    tensor = pb_utils.get_input_tensor_by_name(request, name)
    return torch.from_numpy(tensor.as_numpy())
    # return from_dlpack(tensor.to_dlpack())


def torch2pb(name, tensor):