[2025秋季][T1-1-4] Gary828

[Gary828]

sum topk var var_mean all算子的实现
天数平台有问题，所以没有在其上进行测试

cpu

nvidia

moore

metax

Honor code

HONOR_CODE.md

Reference

REFERENCE.md

其他

topk算子：

• pytorch实现中，value相同时，没有找到固定的逻辑决定哪个indices在前

all算子：

• pytorch针对strided output的outplace有问题，存在输出结果的覆盖现象（一个offset位置的值被更新两次），所以去除了outplace的测试。下例outplace计算结果和inplace不一致

>>> torch.all(input, dim=2, keepdim=True, out=output)
tensor([[[ True],
         [ True],
         [ True],
         [ True],
         [ True]],

        [[ True],
         [ True],
         [ True],
         [ True],
         [False]],

        [[ True],
         [False],
         [ True],
         [ True],
         [ True]],

        [[ True],
         [ True],
         [ True],
         [ True],
         [ True]]])
>>> torch.all(input, dim=2, keepdim=True).bool()
tensor([[[ True],
         [ True],
         [ True],
         [ True],
         [ True]],

        [[ True],
         [ True],
         [ True],
         [ True],
         [False]],

        [[ True],
         [ True],
         [ True],
         [ True],
         [ True]],

        [[ True],
         [ True],
         [ True],
         [ True],
         [ True]]])