如何制作卷积神经网络的对角滤波器？

Question

我用卷积神经网络下棋。在国际象棋中，棋子可以在3x3格内（例如骑士、棋子、国王）、沿着1x8列/8x1档（车和皇后）或沿着对角线（皇后和主教）移动。你知道吗

如果我使用tensorflow，我可以使用tf.keras.layers.Conv2D创建卷积层。我知道如何使过滤器成为3x3块、1x8列和8x1文件，如下所示：

import tensorflow as tf

num_filters = 16
block = tf.keras.layers.Conv2D(num_filters, (3, 3)) # 3x3 block
rank = tf.keras.layers.Conv2D(num_filters, (1, 8)) # filters along a rank
file = tf.keras.layers.Conv2D(num_filters, (8, 1)) # filters along file

我想做一个对角线形状的卷积滤波器/核来展示主教/皇后是如何移动的。这可能吗？也许有办法制作一个8x8卷积滤波器，并将非对角值固定为零？你知道吗