Home / Class/ is_single_direction Class — pytorch Architecture

is_single_direction Class — pytorch Architecture

Architecture documentation for the is_single_direction class in RNN.cpp from the pytorch codebase.

Class cpp

Entity Profile

Source Code

aten/src/ATen/native/mkldnn/RNN.cpp lines 171–176

template<bool is_single_direction>
static std::vector<int64_t> _output_size(const RNNParams& rnn) {
  auto output_channels = is_single_direction ? rnn.hidden_size
                                             : rnn.hidden_size * rnn.num_directions;
  return {rnn.seq_length, rnn.mini_batch, output_channels};
}

Source

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free