test_torch.cpp File Reference

#include <catch2/catch.hpp>
#include <torch/torch.h>
#include <iostream>

Include dependency graph for test_torch.cpp:

Go to the source code of this file.

Functions
int	createNetAndForward (int64_t numInputs, int64_t numNeurons, int64_t numBatch)
	TEST_CASE ("Executing a very basic one layer FNN forward pass", "[PyTorch C++ API]")

Function Documentation

createNetAndForward()

int createNetAndForward	(	int64_t	numInputs,
		int64_t	numNeurons,
		int64_t	numBatch
	)

! Inspired by https://pytorch.org/tutorials/advanced/cpp_frontend.html

Definition at line 39 of file test_torch.cpp.

TEST_CASE()

TEST_CASE	(	"Executing a very basic one layer FNN forward pass"	,
		""	[PyTorch C++API]
	)

Definition at line 67 of file test_torch.cpp.