jittor.contrib

这里是Jittor的贡献代码模块模块的API文档,此模块的代码可能还没有完全成熟,我们将在后续迭代开发中继续完善,您可以通过from jittor import contrib来获取该模块。

jittor.contrib.argmax_pool(x, size, stride, padding=0)[源代码]
jittor.contrib.check(bc)[源代码]
jittor.contrib.concat(arr, dim)[源代码]

Concat Operator can concat a list of jt Var at a specfic dimension.

  • [in] x: input var list for concat

  • [in] dim: concat which dim

  • [out] out: concat result

Example:

jt.concat([jt.array([[1],[2]]), jt.array([[2],[2]])], dim=1)
# return [[1],[2],[2],[2]]
jittor.contrib.setitem(x, slices, value)[源代码]
jittor.contrib.slice_var(x, slices)[源代码]
jittor.contrib.slice_var_index(x, slices)[源代码]