You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

__init__.py 1.7 kB

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. #! /usr/bin/python
  2. # -*- coding: utf-8 -*-
  3. """
  4. TensorLayer provides rich layer implementations trailed for
  5. various benchmarks and domain-specific problems. In addition, we also
  6. support transparent access to native TensorFlow parameters.
  7. For example, we provide not only layers for local response normalization, but also
  8. layers that allow user to apply ``tf.ops.lrn`` on ``network.outputs``.
  9. More functions can be found in `TensorFlow API <https://www.tensorflow.org/versions/master/api_docs/index.html>`__.
  10. """
  11. from .binary_conv import *
  12. from .deformable_conv import *
  13. from .depthwise_conv import *
  14. from .dorefa_conv import *
  15. # from .expert_conv import *
  16. # from .expert_deconv import *
  17. from .group_conv import *
  18. from .quan_conv import *
  19. from .quan_conv_bn import *
  20. from .separable_conv import *
  21. from .simplified_conv import *
  22. # from .simplified_deconv import *
  23. from .super_resolution import *
  24. from .ternary_conv import *
  25. __all__ = [
  26. # simplified conv
  27. 'Conv1d',
  28. 'Conv2d',
  29. 'Conv3d',
  30. # simplified deconv
  31. 'DeConv1d',
  32. 'DeConv2d',
  33. 'DeConv3d',
  34. # expert conv
  35. # 'Conv1dLayer',
  36. # 'Conv2dLayer',
  37. # 'Conv3dLayer',
  38. # expert conv
  39. # 'DeConv1dLayer',
  40. # 'DeConv2dLayer',
  41. # 'DeConv3dLayer',
  42. # atrous
  43. # 'AtrousConv1dLayer',
  44. # 'AtrousConv2dLayer',
  45. # 'AtrousDeConv2d',
  46. # binary
  47. 'BinaryConv2d',
  48. # deformable
  49. 'DeformableConv2d',
  50. # depthwise
  51. 'DepthwiseConv2d',
  52. # dorefa
  53. 'DorefaConv2d',
  54. # group
  55. 'GroupConv2d',
  56. # separable
  57. 'SeparableConv1d',
  58. 'SeparableConv2d',
  59. # subpixel
  60. 'SubpixelConv1d',
  61. 'SubpixelConv2d',
  62. # ternary
  63. 'TernaryConv2d',
  64. #quan_conv
  65. 'QuanConv2d',
  66. 'QuanConv2dWithBN',
  67. ]

TensorLayer3.0 是一款兼容多种深度学习框架为计算后端的深度学习库。计划兼容TensorFlow, Pytorch, MindSpore, Paddle.