Skip to content

Conversation

@kavyasrinet
Copy link

Adding a prototype for one of the functions in layers.py . Once the format and content is approved, I can continue working for all other functions in the file. This PR covers:

  • fully connected layer.
wangkuiyi
wangkuiyi previously approved these changes Nov 8, 2017
Copy link
Collaborator

@wangkuiyi wangkuiyi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

Args:
input: The input tensor to the function
size: The size of the layer
param_attr: The parameters/weights to the FC Layer
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Need to document what are the parameters for param_attr and bias_attr somewhere.

Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Could you elaborate more on what parameters for param_attr would mean ?

@kavyasrinet kavyasrinet deleted the layer_doc branch December 21, 2017 01:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

4 participants