ankane / torch.rb

Deep learning for Ruby, powered by LibTorch
Other
704 stars 30 forks source link

Attr readers for conv nd #28

Closed orlando-labs closed 2 years ago

orlando-labs commented 2 years ago

Hi, @ankane. It is often useful to access convolutional layer attributes, e.g. for output shapes precalculation. So please consider this code update. Thanks.