Go to the source code of this file.
|
| module | tensor3 |
| | Defines a rank-3 tensor.
|
| |
|
| subroutine | tensor3::tensor3_init (t, n1, n2, n3, name) |
| | Initialise a tensor of size n1*n2*n3.
|
| |
| subroutine | tensor3::tensor3_allocate (t, n1, n2, n3) |
| | Allocate a tensor of size n1*n2*n3.
|
| |
| subroutine | tensor3::tensor3_free (t) |
| | Deallocate a tensor.
|
| |
| pure integer function | tensor3::tensor3_size (t) |
| | Returns the number of entries in the tensor.
|
| |
| subroutine | tensor3::tensor3_copy_from (t, memdir, sync) |
| | Easy way to copy between host and device.
|
| |
| pure integer function | tensor3::tensor3_n1 (t) |
| | Returns the size of the first dimension.
|
| |
| pure integer function | tensor3::tensor3_n2 (t) |
| | Returns the size of the second dimension.
|
| |
| pure integer function | tensor3::tensor3_n3 (t) |
| | Returns the size of the third dimension.
|
| |
| subroutine | tensor3::tensor3_assign_tensor3 (t, w) |
| | Assignment \( t = w \).
|
| |
| subroutine | tensor3::tensor3_assign_scalar (t, s) |
| | Assignment \( t = s \).
|
| |
| subroutine | tensor3::tensor3_ptr_init (this, ptr) |
| | Constructor. Just assigns the pointer.
|
| |
| subroutine | tensor3::tensor3_ptr_free (this) |
| | Destructor. Just nullifies the pointer.
|
| |