Pytorch学习（二十一）-自定义C++/ATen扩展
发布时间
阅读量：
阅读量
总说
实在没有办法，
出来混总是要还的，
没有掌握基础代码编写技能就别想混啊。
说实话...简单来说，
很多时候我们需要编写自定义的功能模块，
在Python环境下完成这些操作往往效率不高...
而转而采用CUDA编程语言进行优化，
然后将这些CUDA编译好的模块与Python环境进行链接整合...
方便Python端直接调用使用。
主要是简略拿出官方教程的东西。
根据实践,
补充了一些相关的内容（否则直接阅读官方文档可能会遇到一些难以理解的地方）。
没问题。
看看这个博客,
还可以深入学习。
示例程序
class LLTM(torch.nn.Module):
def __init__(self, input_features, state_size):
super(LLTM, self).__init__()
self.input_features = input_features
self.state_size = state_size
# 3 * state_size for input gate, output gate and candidate cel全部评论 (0)
还没有任何评论哟~