有没有办法迭代并取出张量的3d切片?

2024-10-04 03:21:15 发布

您现在位置:Python中文网/ 问答频道 /正文

有没有一种方法可以迭代3D张量的值进行乘法? 我有一个3x3的遮罩,我想与rgb图像的每个3维像素相乘

一个等价的实现是,假设我有一个3x3张量,它是一个掩码和一个非x200x200x3 rgb图像。我想将3x3掩码乘以(0,0,:)处的(r,g,b)值,然后再乘以(0,1,:)等等,直到(0199,:),然后再乘以(1,0,:)直到(199199,:)生成一个(200200,3)数组。 这是因为每个3x3掩模乘以3x1元素都应该为输入图像中的每个200x200像素产生一个3x1元素(或一个3D像素)

切片在numpy数组中非常容易,因为我可以使用索引进行迭代,但是在Keras中有什么方法可以做到这一点吗? 非常感谢您的帮助 谢谢

Keras.backend.dot能够执行实际的乘法,但是如何实际获取切片呢


Tags: 方法图像numpybackend元素切片rgb像素