不幸的是,我发现如何使用这个函数不是很直观,所以我亲切地问你如何在我的代码中将map_fn转换为while_loop。下面是代码的简化版本: *operations on the 1-D tensors thatnRows))
按照Stackoverflow上的文档和其他问题中的示例,我编写的带有while
我在python输入类似向量(1-D)的函数时有问题,它会被循环。输入应该是list、tuple、numpy.ndarray或类似的输入。重要的是,我将检查输入的长度。因此,typing.Sequence应该是非常适合的。但不是np.array不是Sequence,而是Iterable。Iterable看起来不是正确的选择,因为我需要使用len()。谢谢你的建议。
# arg1 and arg2 s