制作火炬张量的直方图

Question

我有一个大小为 (100*21) 的火炬张量，其值介于 0 和 1 之间。 我如何制作类似于所附图像中的直方图，其中蓝色指的是最小值，红色指的是最大值。