搜索 - 腾讯云开发者社区-腾讯云

文章/答案/技术大牛

发布

来自专栏计算机视觉理论及其实现
tf.scatter_nd()
For example, if we wanted to insert two slices in the first dimension of a rank-3 tensor with two matrices
1.7K10编辑于 2022-09-03
来自专栏信数据得永生
FastAI 之书（面向程序员的 FastAI）（二）
描述这样的张量最常见的方式是称之为rank-3 张量。我们经常需要将集合中的单个张量堆叠成一个张量。不出所料，PyTorch 带有一个名为stack的函数，我们可以用它来实现这个目的。我们通过沿着我们堆叠的 rank-3 张量的维度 0 取平均值来计算所有图像张量的平均值。这是索引所有图像的维度。换句话说，对于每个像素位置，这将计算所有图像中该像素的平均值。
72120编辑于 2024-02-17