沿更深的维度更新张量值

Question

我具有MxNxC形状的张量A，其中M代表示例数量，N代表特征数量，C代表3欧拉旋转角。我也有一个类似形状的张量B，但是没有角度，而是坐标。 需要的是将这两个张量都转换为一个包含仿射变换矩阵的张量，以便其形状类似于MxNx4x4。我不知道如何一起遍历这些张量，我一直在寻找tf.map_fn和tf.scan，但它们仅在第一维上进行迭代。我正在寻找的是一种方法，可以将以下功能应用于最后一个轴上的所有元素。

def f(angles, vector): #dimensions 3 or 3x1
    ...
    return matrix # dimension 4x4

任何帮助都会有用，谢谢！

Answer 1

您可以尝试以下操作：

<script src="https://www.gstatic.com/charts/loader.js"></script>
<div id="timeline-tooltip"></div>

这应该可以解决问题！