Python shape示例

示例#1

文件： __init__.py 项目： whztt07/plaidml

def shape(x):
    return _KerasNode('shape', tensor=edsl.shape(x.tensor))

示例#2

def shape(x):
    logger.debug('shape(x: {})'.format(x))
    return _KerasNode('shape', tensor=edsl.shape(x.tensor))

示例#3

def shape(x):
    ret = _KerasNode('shape', tensor=edsl.shape(x.tensor))
    # Save the TensorDims directly on the _KerasNode, where they can be extracted if needed
    ret._RawTensorDims = edsl.TensorDims(x.tensor.shape.ndims)
    x.tensor.bind_dims(*ret._RawTensorDims)
    return ret