入门指南
自带
扩展 PyKEEN
参考
pykeen.nn
附录
基础类:ConcatCombination
ConcatCombination
通过连接后沿同一轴进行聚合来组合表示。
初始化组合。
aggregation (str | Callable[[Tensor], Tensor] | None) – 聚合或其提示,参见 class_resolver.contrib.torch.aggregation_resolver
class_resolver.contrib.torch.aggregation_resolver
dim (int) – 连接和缩减的维度。
方法总结
forward(xs)
forward
组合一系列单独的表示。
iter_extra_repr()
iter_extra_repr
遍历extra_repr()的组件。
extra_repr()
方法文档
xs (Sequence[Tensor]) – 形状: (*batch_dims, *input_dims_i) 单个表示
形状: (*batch_dims, *output_dims) 一个组合表示
Tensor
此方法通常被重写。一个常见的模式是
def iter_extra_repr(self) -> Iterable[str]: yield from super().iter_extra_repr() yield "<key1>=<value1>" yield "<key2>=<value2>"
一个可迭代的extra_repr()的各个组件
Iterable[str]