New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Use F.matmul instead of F.batch_matmul #141
Conversation
Oops, the PR depends on the new F.matmul in v3. |
I think It is better to support both v2 and v3 for now by checking the version of chainer and use F.numpy_like_matmul for v2. |
Can you merge the current master to this PR so that CI runs with both Chainer v2 and v3? |
chainerrl/misc/chainer_compat.py
Outdated
pkg_resources.get_distribution("chainer").version) | ||
|
||
if chainer_version < StrictVersion('3.0.0a1'): | ||
def matmul_v3(a, b, **kwargs): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you add a docstring to explain why this is needed?
Rebased to the current master |
LGTM |
batch_matmul is going to be unavailable in Chainer v3, or would be deprecated even if chainer/chainer#3016 is merged.