首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

tf.lite

“index_override”要使用的全局索引。这对应于将生成的最终存根中的参数顺序。返回值:被包裹的输入张量。...“index_override”要使用的全局索引。这对应于将生成的最终存根中的参数顺序。返回值:缠绕输出张量。...* * kwargs:明白了返回值:包装输出(具有附加元数据的标识替代)。这些也是tf.Tensor。...这必须是一个可调用的对象,返回一个支持iter()协议的对象(例如一个生成器函数)。生成的元素必须具有与模型输入相同的类型和形状。八、tf.lite.TargetSpec目标设备规格。...这用于将TensorFlow GraphDef或SavedModel转换为TFLite FlatBuffer或图形可视化。属性:inference_type:输出文件中实数数组的目标数据类型。

5.3K60
  • 您找到你想要的搜索结果了吗?
    是的
    没有找到

    tensorflow中的slim函数集合

    参数:作用域:筛选要返回的变量的可选作用域。后缀:用于过滤要返回的变量的可选后缀。返回值:集合中具有范围和后缀的变量列表。...第n个维度需要具有指定数量的元素(类的数量)。参数:logits: N维张量,其中N > 1。scope:variable_scope的可选作用域。返回值:一个形状和类型与logits相同的“张量”。...只支持浮点类型返回值:生成单位方差张量的初始化器可能产生的异常:ValueError: if `dtype` is not a floating point type.TypeError: if `mode...[batch_size,…]outputs_collections:用于添加输出的集合scope:name_scope的可选作用域返回值:一个具有形状[batch_size, k]的平坦张量。...支持' NHWC '(默认值)和' NCHW 'outputs_collections:将输出添加到其中的集合scope:name_scope的可选作用域返回值:表示池操作结果的“张量”可能产生的异常:

    1.6K30

    tf.train.batch

    参数张量可以是张量的列表或字典。函数返回的值与张量的类型相同。这个函数是使用队列实现的。队列的QueueRunner被添加到当前图的QUEUE_RUNNER集合中。...注意: 如果dynamic_pad为False,则必须确保(i)传递了shapes参数,或者(ii)张量中的所有张量必须具有完全定义的形状。如果这两个条件都不成立,将会引发ValueError。...在这种情况下,对于每个加入值为None的维度,其长度可以是可变的;在退出队列时,输出张量将填充到当前minibatch中张量的最大形状。对于数字,这个填充值为0。对于字符串,这个填充是空字符串。...此外,通过shape属性访问的所有输出张量的静态形状的第一个维度值为None,依赖于固定batch_size的操作将失败。参数:tensors: 要排队的张量列表或字典。...返回值:与张量类型相同的张量列表或字典(除非输入是一个由一个元素组成的列表,否则它返回一个张量,而不是一个列表)。

    1.4K10

    tf.train

    boundaries: 张量、int或浮点数的列表,其条目严格递增,且所有元素具有与x相同的类型。values: 张量、浮点数或整数的列表,指定边界定义的区间的值。...它应该比边界多一个元素,并且所有元素应该具有相同的类型。name: 一个字符串。操作的可选名称。默认为“PiecewiseConstant”。返回值:一个0维的张量。...(如果为真)允许从保存文件中还原变量,其中变量具有不同的形状,但是相同数量的元素和类型。...此外,通过shape属性访问的所有输出张量的静态形状的第一个维度值为None,依赖于固定batch_size的操作将失败。参数:tensors: 要排队的张量列表或字典。...返回值:与张量类型相同的张量列表或字典(除非输入是一个由一个元素组成的列表,否则它返回一个张量,而不是一个列表)。

    3.6K40

    tf.variable_scope

    请注意,在1.0版本之前和包括1.0版本之前,允许(尽管明确地不鼓励)将False传递给重用参数,从而产生了与None略有不同的无文档化行为。...从1.1.0开始传递None和False,因为重用具有完全相同的效果。关于在多线程环境中使用变量作用域的注意事项:变量作用域是线程本地的,因此一个线程不会看到另一个线程的当前作用域。...dtype: 在此范围中创建的变量的类型(默认为传递的范围中的类型,或从父范围继承的类型)。use_resource: 如果为False,则所有变量都是常规变量。...如果为真,则使用具有定义良好语义的实验性资源变量。默认值为False(稍后将更改为True)。当启用紧急执行时,该参数总是强制为真。...constraint: 优化器更新后应用于变量的可选投影函数(例如,用于为层权重实现规范约束或值约束)。函数必须将表示变量值的未投影张量作为输入,并返回投影值的张量(其形状必须相同)。

    2K20

    NumPy 1.26 中文官方指南(三)

    方便的属性 array具有.T 属性,返回数据的转置。 matrix还具有.H、.I 和.A 属性,分别返回矩阵的共轭转置、逆矩阵和 asarray()。...便利属性 array具有.T 属性,返回数据的转置。 matrix还具有.H, .I 和 .A 属性,分别返回矩阵的共轭转置、逆和asarray()。...返回外部对象 第三种特性集旨在使用 NumPy 函数实现,然后将返回值转换为外部对象的实例。...警告 尽管将 ndarrays 和张量混合使用可能很方便,但不建议这样做。它对于非 CPU 张量不起作用,在一些边缘情况下会有意外的行为。用户应该优先显式地将 ndarray 转换为张量。...这不是最佳的,因为将数组强制转换为 ndarrays 可能会导致性能问题或创建副本和元数据丢失,因为原始对象及其可能具有的任何属性/行为都会丢失。

    38310

    tf.Session

    返回的可调用函数将接受len(feed_list)参数,其类型必须与feed_list的各个元素的提要值兼容。例如,如果feed_list的元素i是tf。...张量,返回的可调用的第i个参数必须是一个numpy ndarray(或可转换为ndarray的东西),它具有匹配的元素类型和形状。...返回的可调用函数将具有与tf.Session.run(fetches,…)相同的返回类型。例如,如果fetches是tf。张量,可调用的将返回一个numpy ndarray;如果fetches是tf。...feed_dict中的每个键都可以是以下类型之一:如果键是tf.Tensor,其值可以是Python标量、字符串、列表或numpy ndarray,可以转换为与该张量相同的dtype。...如果键是张量或稀疏张量的嵌套元组,则该值应该是嵌套元组,其结构与上面映射到其对应值的结构相同。feed_dict中的每个值必须转换为对应键的dtype的numpy数组。

    2.7K20

    TorchScript使用的注意事项和常见错误

    Pytorch1.3新出了移动端部署功能,想尝试一下,目前已将除安卓端部署以外的流程走通,但是因为pytorch量化不太好用,目前已经转向研究ONNX。...函数的默认参数如果不是tensor的话,需要指定类型; 6. list中元素默认为tensor,如果不是,也要指定类型; 7. tensor.bool()操作不支持,可以直接用tensor>0来替代;...在TorchScript中,有一种Optional类型,举例:在一个函数中,如果可以通过if控制来返回None或者tensor,那么这个返回值会被认定为Optional[Tensor],这会导致无法对该返回值使用...TorchScript中带有梯度的零维张量无法当做标量进行计算,这个问题可能会在使用C++自定义TorchScript算子时遇到。...常见错误 ValueError: substring not found forward函数中不允许出现中文注释 Module is not iterable(大概是这样的错误) 不支持模型遍历及对模型取下标的操作

    2K40

    tf.while_loop

    body是一个可调用的变量,返回一个(可能是嵌套的)元组、命名元组或一个与loop_vars具有相同特性(长度和结构)和类型的张量列表。...稀疏张量和转位切片的形状不变式特别处理如下:a)如果一个循环变量是稀疏张量,那么形状不变量必须是张量形状([r]),其中r是由稀疏张量表示的稠密张量的秩。...这意味着稀疏张量的三个张量的形状是([None], [None, r], [r])。注意:这里的形状不变量是SparseTensor.dense_shape属性的形状。它一定是向量的形状。...这些张量是内存消耗的主要来源,在gpu上进行训练时经常导致OOM错误。当swap_memory标志为true时,我们将这些张量从GPU交换到CPU。例如,这允许我们用很长的序列和大量训练RNN模型。...name:返回的张量的可选名称前缀。返回值:循环变量的输出张量。返回值具有与loop_vars相同的结构。

    2.8K40
    领券