Function _create_tensor

python/paddle/base/framework.py:1571–1592

(
    type=core.VarDesc.VarType.DENSE_TENSOR,
    name=None,
    shape=None,
    dtype=None,
    persistable=None,
    **kwargs,
)

def _create_tensor(
    type=core.VarDesc.VarType.DENSE_TENSOR,
    name=None,
    shape=None,
    dtype=None,
    persistable=None,
    **kwargs,
):
    if dtype is not None:
        dtype = convert_to_proto_type(dtype)
    else:
        dtype = core.VarDesc.VarType.FP32

    eager_tensor = core.eager.Tensor(
        dtype,
        list(shape) if shape else [],
        name,
        type if type else core.VarDesc.VarType.DENSE_TENSOR,
        True if persistable else False,
    )
    eager_tensor.retain_grads()
    return eager_tensor


def _all_is_type(vals, expected_type):