tf.nn函数总结

1、tf.nn.conv2d

conv2d(
    input,
    filter,
    strides,
    padding,
    use_cudnn_on_gpu=None,
    data_format=None,
    name=None
)

根据给定的四维张量和滤波张量计算2-D滤波。其中,给定输入的张量维度为[batch, in_height, in_weidht, in_channels]和滤波核的张量的维度为[filter_height, filter_width, in_channesl, out_channels],计算会转换为一下步骤

(1)将filter平铺成二维的张量,维度为[filter_height*fliter_width*in_channels, output_channels]

(2)将image转换成一个张量[batch, out_height, out_width, filter_height*filter_width*in_channels]

(3)对于每个patch,右乘滤波矩阵和image块

默认格式为:NHWC
output[b, i, j, k] =
    sum_{di, dj, q} input[b, strides[1] * i + di, strides[2] * j + dj, q] *
                    filter[di, dj, q, k]

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值