如何解决如何在张量流中将矩阵批次乘以相应的张量1 D
假设我有形状为(3,2,2)的张量,并且我想将每个(2,2)矩阵乘以某个1d张量。这是代码:
a=tf.constant([[[1,0],[1,0]],[[2,[[3,0]] ])
b=tf.constant([1,3])
在这里,我想将[[1,0]]
乘以1,[[2,0]]
乘以2,将[[3,0]]
乘以3。因此,在乘法之后,我应该得到:
[[[1,[[4,[2,[[9,[3,0]] ]
编辑:我不使用急切执行
解决方法
tf.convert_to_tensor([tf.multiply(aa,bb) for aa,bb in zip(a,b)])
<tf.Tensor: shape=(3,2,2),dtype=int32,numpy=
array([[[1,0],[1,0]],[[4,[2,[[9,[3,0]]])>
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。