grain ~master (2021-05-15T05:53:16Z)

Convolution.outShape

grain functions binary Convolution

https://pytorch.org/docs/master/nn.html#convolution-layers

struct Convolution(T, size_t imDims, bool isConv = false, bool isNchw = true)
auto
outShape
(
uint[nbDims] inShape
,
uint[nbDims] weightShape
)

Meta

Source

See Implementation

grain functions binary Convolution aliasesfunctionsmixinsvariables

Page generated by adrdox