阅读(4.5k) 书签 (0)

TensorFlow函数教程:tf.nn.quantized_conv2d

2019-01-31 13:51 更新

tf.nn.quantized_conv2d函数

tf.nn.quantized_conv2d(
    input,
    filter,
    min_input,
    max_input,
    min_filter,
    max_filter,
    strides,
    padding,
    out_type=tf.qint32,
    dilations=[1, 1, 1, 1],
    name=None
)

请参阅指南:神经网络>候选采样

在给定量化4D输入和滤波器张量的情况下计算2D卷积.

输入是量化张量,其中最低值表示相关最小值的实数,最高值表示最大值.这意味着您只能通过考虑返回的最小值和最大值来以相同的方式解释量化输出.

参数:

  • input:一个Tensor,必须是下列类型之一:qint8,quint8,qint32,qint16,quint16.
  • filter:一个Tensor,必须是下列类型之一:qint8,quint8,qint32,qint16,quint16.filter的input_depth维度必须与输入的深度维度匹配.
  • min_input:一个Tensor,类型为float32.最小量化输入值表示的浮点值.
  • max_input:一个Tensor,类型为float32.最大量化输入值表示的浮点值.
  • min_filter:一个Tensor,类型为float32.最小量化过滤器值表示的浮点值.
  • max_filter:一个Tensor,类型为float32.最大量化过滤器值表示的浮点值.
  • strides:ints列表.输入张量的每个维度的滑动窗口的步幅.
  • padding:string,可以是:"SAME", "VALID".要使用的填充算法的类型.
  • out_type:可选的tf.DType,可以是:tf.qint8, tf.quint8, tf.qint32, tf.qint16, tf.quint16.默认为tf.qint32.
  • dilations:可选ints的列表.默认为[1, 1, 1, 1].长度为4的1-D张量.input每个维度的膨胀系数.如果设置为k> 1,则该维度上的每个滤镜元素之间将有k-1个跳过的单元格.维度顺序由值data_format确定.批次和深度尺寸的扩张必须为1.
  • name:操作的名称(可选).

返回:

Tensor对象的元组(output,min_output,max_output).

  • output:一个Tensor,类型为out_type.
  • min_output:一个Tensor,类型为float32.
  • max_output:一个Tensor,类型为float32.