如何从HxWxD张量中获得D平均值

Question

如何在Deeplearnjs中创建一个图形元素，将我的[h，w，d]形状张量转换为[d]形状，其中每个都是该层的最大值。如果h和w相同，则可以使用maxpool函数完成。如果像是一样的意思。使用conv2d可以实现平均值，但仅当w和h相等时才能实现。

我需要在图表中使用它，以便我可以应用培训。

Answer 1

您可以执行dl.mean(your_tensor, [0, 1])或your_tensor.mean([0, 1])来获得h和w维度的平均值。任何一个都会返回一个形状为[d]的张量。这也适用于训练，因为deeplearnjs已转移到急切的执行模式，并在mean reduction op中定义了渐变。您可以在没有图表的情况下查看mnist_eager演示以获取培训示例。