(equation, *operands)
return torch.einsum(equation, operands)
Variadic version of torch.einsum to match numpy api.
Variadic version of torch.einsum to match numpy api.
Variadic version of torch . einsum to match numpy api .
def einsum(equation, *operands):
    """Variadic version of torch.einsum to match numpy api."""
    # rename symbols to support PyTorch 0.4.1 and earlier,
    # which allow only symbols a-z.
    equation = convert_to_valid_einsum_chars(equation)

    torch, _ = _get_torch_and_device()
    return torch.einsum(equation, operands)
def einsum ( equation , * operands ) : # rename symbols to support PyTorch 0.4.1 and earlier, # which allow only symbols a-z. equation = convert_to_valid_einsum_chars ( equation ) torch , _ = _get_torch_and_device ( ) return torch . einsum ( equation , operands )