blace::ops::SumOp Class Reference

#include <public_ops.h>

Inheritance diagram for blace::ops::SumOp:

Public Member Functions
	SumOp (blace::ops::OpP input, std::vector< int64_t > sizes, bool keepdim)

Detailed Description

Sums tensors along given dimensions.

Constructor & Destructor Documentation

SumOp()

blace::ops::SumOp::SumOp	(	blace::ops::OpP	input,
		std::vector< int64_t >	sizes,
		bool	keepdim
	)

Sums tensor along dimensions.

Parameters

input
sizes	List of dimensions to sum along.
keepdim	Keep sumed dimensions?