我使用tensorflow的问题如下:
对于两个张量
[[x11,x12...],[x21,x22...],...[xn1,xn2...]]
和
[y1,y2,...yn],
我想沿轴0乘以它们得到
[[x11*y1,x12*y1...],[x21*y2,x22*y2...]...]
例如,对于 [[1,2],[3,4]]
和 [1,2]
,我想得到结果tensor [[1,2],[6,8]]
.
真实情况是我有两个张量A和B形状(批次,高度,宽度,n_channels)和(批次,1) . 两者都是张量流中定义的张量 . 对于批处理中A的每个图像,我想将它与B中的相应值相乘 .
1 回答
给定一个二维张量
x
和一个向量y
,你只需要做:或者,如果你更喜欢它: