Tensor的Broadcasting操作

针对tensorflow2.0版本

Broadcasting的主要特点:expand without copying data(扩展了维度,但是不复制数据,不占用内存空间)

(1)隐式的Broadcasting

tensor的一些操作符已经包含了Broadcasting,比如下面这个:

In [3]: a = tf.random.normal([4,28,28,3])

In [4]: b = tf.random.normal([3])

In [5]: (a+b).shape
Out[5]: TensorShape([4, 28, 28, 3])

随机两个矩阵a,b,但矩阵维度不一致,当a+b时,系统会自动扩展b的维度(不复制数据,不增加内存),以能够与a相加。

(2)显示的Broadcasting(tf.broadcast_to)

In [6]: c = tf.broadcast_to(tf.random.normal([4,1,1,1]),[4,28,28,3])

In [7]: c.shape
Out[7]: TensorShape([4, 28, 28, 3])

使用tf.broadcast_to函数手动使c的维度扩展。

(3)Broadcasting与tf.tile的区别

功能相近,但是tf.tile是复制数据,需占用内存,Broadcasting不用。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值